from albumentations.pytorch.transforms import ToTensorV2
我使用了上面的代码,但它不起作用。
只需添加一个带有以下行的代码块
! pip install albumentations==0.4.6
在您进行导入的块上方。我尝试在没有特定版本的情况下安装它,但失败了。
当我没有在 pip install 中指定版本号时,安装了 0.1.12 版本,其中不包含 ToTensorV2。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)