解决方法：
更改语句
from albumentations.pytorch import ToTensorV2
在代码中，将出现ToTensor() 的地方都改成ToTensorV2()
from albumentations.pytorch import ToTensor 报错：在 ‘__init__.py‘ 中找不到引用 ‘ToTensor‘
猜你喜欢
转载自blog.csdn.net/weixin_44025103/article/details/130748992
今日推荐
周排行