无法在 Colab 中导入 ToTensorV2

Question

from albumentations.pytorch.transforms import ToTensorV2

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

Answer 1

只需添加一个带有该行的代码块

! pip install albumentations==0.4.6

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

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