search

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.61k stars 303 forks source link

pytorch版本训练的版本咨询 #218

Open WMX1007 opened 8 months ago

WMX1007 commented 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

lyuwenyu commented 8 months ago

是的 版本问题 详细参考：https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/requirements.txt

lixinru77 commented 8 months ago

升级就行：conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia