from albumentations.pytorch.transforms import ToTensorV2
我使用了上面的代码,但是它不起作用。
只需添加一行代码
!点安装专辑== 0.4.6
在您执行导入的块上方。我尝试在没有特定版本的情况下安装它,但失败了。
当我未在pip安装中指定版本号时,安装的版本0.1.12不包含ToTensorV2。
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
点击生成二维码
我来说两句