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.31k stars 258 forks source link

torchvision error #398

Open lime-s opened 1 month ago

lime-s commented 1 month ago

rtdetrv2_pytorch/src/data/_misc.py 中需要注意可以将原语句改为importlib.metadata.version('torchvision')[:6] == '0.15.2':否则如果你的torchvision为0.15.2+cu118会报错