无法修复ImportError:无法从Colab上的“albumentations.pytorch”导入名称“ToTensorV2”

xuo3flqw  于 2023-10-20  发布在  其他
关注(0)|答案(3)|浏览(173)

我尝试了this question的所有三种解决方案来解决这个问题,但仍然收到一个错误:

ImportError: cannot import name 'ToTensorV2' from 'albumentations.pytorch' (/usr/local/lib/python3.7/dist-packages/albumentations/pytorch/__init__.py)

当前代码:

!pip install -U albumentations
import albumentations 
from albumentations.pytorch import ToTensorV2
baubqpgj

baubqpgj1#

你试过工厂重置运行时吗?如果你没有,也许python仍然在加载早期版本。我刚刚在一个新的colab示例上测试了链接问题的公认解决方案,它工作得很好:

!pip install albumentations==0.4.6
import albumentations 
from albumentations.pytorch import ToTensorV2
hlswsv35

hlswsv352#

试试这个,应该可以解决你的问题:

from albumentations.pytorch.transforms import ToTensor, ToTensorV2
holgip5t

holgip5t3#

在这里它为我工作。github的插件没有任何问题。
!pip install -U git+https://github.com/albu/albumentations > /dev/null && echo import albumentations from albumentations.pytorch import ToTensorV2

相关问题