关于使用pycocotools验证

ouyanghaodong / DEYO

Implementation of paper - DEYO: DETR with YOLO for End-to-End Object Detection

GNU Affero General Public License v3.0

72 stars 7 forks source link

Open Letitialky opened 4 months ago

Letitialky commented 4 months ago

请问作者怎么使用pycocotools对模型进行验证？有没有生成groud truth和predictions的json的代码呢？

ouyanghaodong commented 4 months ago

# Eval the model
model = RTDETR("DEYO-tiny.pt")
model.val(data = "coco.yaml")  # for DEYO-tiny: 37.6 AP