Google Colab ToTensor2 albumentations

vision
1

Hello, I tried to import ToTensor2 as a “from albumentations import ToTensor2” in colab, but it shows an error "cannot import name ‘ToTensor2’ from ‘albumentations’ " Could someone help?

1 Like
2

I guess you have a typo and want to use ToTensorV2?

3

Yes, but still i cannot fix it.

4

Hello, is there any solution for this problem?
I have tried

!pip install -q -U albumentations

import albumentations 
from albumentations.pytorch import ToTensorV2

and

!pip install albumentations==1.1.0
import albumentations 
from albumentations.pytorch import ToTensorV2

both of them do not work. Is there any other way to use albumenations ?

Thank you

5

Could you describe your issue with more details please, as:

!pip install albumentations
import albumentations
from albumentations.pytorch import ToTensorV2
albumentations.__version__
> '1.1.0'

works in my setup.

1 Like