代码:
from albumentations.pytorch import ToTensorV2, ToTensor
如果直接pip install albumentations会安装最新版本,但最新版本的albumentations去掉了ToTensorV2.
因此需要安装指定版本,先uninstall,然后执行:
pip install albumentations==0.4.6
代码:
from albumentations.pytorch import ToTensorV2, ToTensor
如果直接pip install albumentations会安装最新版本,但最新版本的albumentations去掉了ToTensorV2.
因此需要安装指定版本,先uninstall,然后执行:
pip install albumentations==0.4.6