无法在Colab中导入ToTensorV2

Question

from albumentations.pytorch.transforms import ToTensorV2

我使用了上面的代码，但它不起作用。

Answer 1

只需添加一行代码块即可。

! pip install albumentations==0.4.6

在您进行导入的块之上。我试着在没有特定版本的情况下安装它，结果失败了。

当我没有在pip中指定版本号时，就安装了不包含ToTensorV2的0.1.12版本。

Answer 2

确保您有最新版本

!pip install --upgrade --force-reinstall --no-deps albumentations

Answer 3

从吉特布那里得到消息。Colab示例的用法

!pip install -U git+https://github.com/albu/albumentations > /dev/null