安装albumentation时利用conda进行的安装
安装步骤是:
conda install -c conda-forge imgaug
conda install -c conda-forge albumentations
python3.6 main.py在命令行运行时报错:
解决方法:
找到data.py文件把from albumentations.pytorch import TotensorV2修改为
from albumentations.pytorch.transforms import ToTensor
运行成功了