No module named 'albumentations.torch.functional'

[rknoopsLime]

in demo.ipynb, the line

from dataset import load_image

gives

ModuleNotFoundError: No module named 'albumentations.torch.functional'

I used conda to install albumentations, and the ipnyb command 'import albumentations.torch' loads properly. Apologies if this is an error from my side.

[EmmaSRH]

in demo.ipynb, the line

from dataset import load_image

gives

ModuleNotFoundError: No module named 'albumentations.torch.functional'

I used conda to install albumentations, and the ipnyb command 'import albumentations.torch' loads properly. Apologies if this is an error from my side.

I solved it by "sudo pip3 install albumentations".

[sakares]

Depends on the albementations version you have used. you may want to workaround by replacing from

from albumentations.torch.functional import img_to_tensor

to

from albumentations.pytorch.transforms import img_to_tensor

[woctezuma]

The issue can arise because requirements.txt does not specify the version of albumentations to use:

robot-surgery-segmentation/requirements.txt

Lines 1 to 4 in de12469

	torch==0.4.0
	numpy==1.14.0
	tqdm==4.19.4
	albumentations

Following your remark about version, I have run pip install --upgrade albumentations, and this has solved the issue.

[headingy]

albumentations = "0.5.2" would be the highest version that compiles