Не удается импортировать ToTensorV2 в Colab

Question

from albumentations.pytorch.transforms import ToTensorV2

Я использовал приведенный выше код, но он не работает.

Answer 1

Просто добавьте блок кода со строчкой

! pip install albumentations == 0.4.6

над блоком, в котором вы выполняете импорт. Я попытался установить его без указанной c версии, и это не удалось.

Когда я не указал номер версии в pip install, была установлена ​​версия 0.1.12, которая не содержит ToTensorV2.