13717630148
commented
3 years ago pip install albumentations==0.4.6就可以了
Open 13717630148 opened 3 years ago
13717630148
commented
3 years ago pip install albumentations==0.4.6就可以了
train_parsing.py时遇到报错 Traceback (most recent call last): File "train_parsing.py", line 2, in
from dataset import VeRi3kParsingDataset, get_preprocessing, get_training_albumentations, get_validation_augmentation
File "/data/yijinhui/PVEN/examples/parsing/dataset.py", line 11, in
from vehicle_reid_pytorch.data.transforms import AlbuRandomErasing
File "/data/yijinhui/.local/lib/python3.6/site-packages/vehicle_reid_pytorch-1.0-py3.6.egg/vehicle_reid_pytorch/data/init.py", line 2, in
File "/data/yijinhui/.local/lib/python3.6/site-packages/vehicle_reid_pytorch-1.0-py3.6.egg/vehicle_reid_pytorch/data/demo_transforms.py", line 2, in
ImportError: cannot import name 'ToTensor'
搜了下应该是albumentations版本的问题，我的是1.1.0，想问下超哥的