无法在Colab中导入ToTensorV2

Question

from albumentations.pytorch.transforms import ToTensorV2

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

Answer 1

只需在代码行中添加一个代码块

！ pip install albumentations == 0.4.6

在执行导入的块上方。我尝试在没有特定版本的情况下安装它，但失败了。

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

Answer 2

确保您拥有最新版本

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

Answer 3

从Github获取专辑。 Usage on Colab Example

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