(已解决报错)ImportError: cannot import name ‘ToTensor’

代码:

from albumentations.pytorch import ToTensorV2, ToTensor

如果直接pip install albumentations会安装最新版本,但最新版本的albumentations去掉了ToTensorV2.

因此需要安装指定版本,先uninstall,然后执行:

pip install albumentations==0.4.6

你可能感兴趣的:(调试代码的一些错误解决,python,深度学习)