Impossible d'importer ToTensorV2 dans Colab

Question

from albumentations.pytorch.transforms import ToTensorV2

J'ai utilisé le code ci-dessus, et cela ne fonctionne pas.

Answer 1

Il suffit d'ajouter un bloc de code avec la ligne

! pip install albumentations==0.4.6

au-dessus du bloc où vous effectuez l'importation. J'ai essayé de l'installer sans la version spécifique et ça a échoué.

Lorsque je n'ai pas spécifié le numéro de version dans pip install, la version 0.1.12 a été installée qui ne contient pas ToTensorV2.

Answer 2

Assurez-vous que vous avez la dernière version

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

Answer 3

Obtenez des albuminations à partir de Github. Utilisation sur Colab Exemple

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