Open WHK1229 opened 3 months ago
You should modify coco_eval
file
https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetrv2_pytorch/src/data/dataset/coco_eval.py
You can use torch.profiler
直接在命令行运行python /home/whk/hdd/RT-DETR-main/rtdetr_pytorch/src/data/coco/coco_eval.py 没有反应,也不报错
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
How to output datasets with evaluation accuracy of P, R, mAP, FPS, Gflops