【发布时间】:2020-11-02 10:41:54
【问题描述】:
from albumentations.pytorch.transforms import ToTensorV2
我用了上面的代码,还是不行。
【问题讨论】:
标签: pytorch google-colaboratory
from albumentations.pytorch.transforms import ToTensorV2
我用了上面的代码,还是不行。
【问题讨论】:
标签: pytorch google-colaboratory
只需添加一行代码块
! pip install albumentations==0.4.6
在您进行导入的块上方。我尝试在没有特定版本的情况下安装它,但失败了。
当我没有在 pip install 中指定版本号时,安装了不包含 ToTensorV2 的版本 0.1.12。
【讨论】:
确保您拥有最新版本
!pip install --upgrade --force-reinstall --no-deps albumentations
【讨论】:
从 Github 获取专辑。 Usage on Colab Example
!pip install -U git+https://github.com/albu/albumentations > /dev/null
【讨论】: