Open WMX1007 opened 8 months ago
我使用的是自己的数据集,执行train.py时总是报错ImportError: cannot import name 'datapoints' from 'torchvision' AttributeError: module 'torchvision.transforms' has no attribute 'RandomPhotometricDistort'等 查了一下没有相关错误解析,不知道是不是版本原因,我是用的是2.2.0版本的torch,0.17.0版本的torchvision
是的 版本问题 详细参考:https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/requirements.txt
升级就行:conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
我使用的是自己的数据集,执行train.py时总是报错ImportError: cannot import name 'datapoints' from 'torchvision' AttributeError: module 'torchvision.transforms' has no attribute 'RandomPhotometricDistort'等 查了一下没有相关错误解析,不知道是不是版本原因,我是用的是2.2.0版本的torch,0.17.0版本的torchvision