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.64k stars 305 forks source link

Verification and evaluation of code #403

Open WHK1229 opened 3 months ago

WHK1229 commented 3 months ago

How to output datasets with evaluation accuracy of P, R, mAP, FPS, Gflops

lyuwenyu commented 3 months ago
  1. You should modify coco_eval file

    https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetrv2_pytorch/src/data/dataset/coco_eval.py

  2. You can use torch.profiler

    https://pytorch.org/docs/stable/profiler.html

WHK1229 commented 3 months ago

直接在命令行运行python /home/whk/hdd/RT-DETR-main/rtdetr_pytorch/src/data/coco/coco_eval.py 没有反应,也不报错

lyuwenyu commented 3 months ago

CUDA_VISIBLE_DEVICES=0,1,2,3 torchrun --master_port=9909 --nproc_per_node=4 tools/train.py -c path/to/config -r path/to/checkpoint --test-only