tinyvision / DAMO-YOLO

DAMO-YOLO: a fast and accurate object detection method with some new techs, including NAS backbones, efficient RepGFPN, ZeroHead, AlignedOTA, and distillation enhancement.
Apache License 2.0
3.75k stars 470 forks source link

单类训练finetune自己的数据集,DAMO-YOLO-S远不及YOLOX-S ? #106

Open woxiaohan opened 1 year ago

woxiaohan commented 1 year ago

Before Asking

Search before asking

Question

您好,我使用自己的数据在damo-yolo s上finetune(damoyolo_tinynasL25_S_460.pth) 对比训练yolox s网络. 单GPU batch-size改为16, 训练300个epoch,这与yolox 的设定一致,其余训练超参数没有改变; nms_conf_thre和nms_iou_thre也改为了和yolox 一致,分别设置为0.01和0.65。 但是最终在val上的mAP0.50:0.95 为0.75,相对于yolox s的0.87低了很多,这与https://www.modelscope.cn/models/damo/cv_tinynas_object-detection_damoyolo/summary中的模型评测相对值相差较大. 是否damo yolo模型对单类训练有其他要求? 类似的问题在https://github.com/tinyvision/DAMO-YOLO/issues/77中也没找出合理解释.

Additional

No response