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 259 forks source link

pytorch测试问题 #258

Open zzyy410 opened 5 months ago

zzyy410 commented 5 months ago

您好，请问pytorch版本代码训练后的模型如何进行test？按照GitHub所给命令torchrun --nproc_per_node=4 tools/train.py -c configs/rtdetr/rtdetr_r50vd_6x_coco.yml -r path/to/checkpoint --test-only似乎使用的仍然是val过程中使用的数据集，test过程是不是应该使用训练时候未使用的数据集？

lyuwenyu commented 5 months ago

你说的test可以参考deploy的代码