search

ouyanghaodong / DEYO

Implementation of paper - DEYO: DETR with YOLO for End-to-End Object Detection
GNU Affero General Public License v3.0
61 stars 5 forks source link

关于验证的问题 #20

Open Letitialky opened 1 month ago

Letitialky commented 1 month ago

在单独进行模型验证时，yolov5、yolov8等yolo系列模型在验证时可以设置iou和conf的阈值，不用阈值下的验证结果略有差异。我注意到DEYO与yolov8的代码结构大致相同，在单独验证时也可以设置iou和conf，但是我在训练自己的数据集后，运行model.val()并设置iou和conf，发现设置iou=0.5和iou=0.7的验证结果完全相同，请问作者是什么原因呢？DEYO的验证结果与不同的iou和conf有关吗？

ouyanghaodong commented 1 month ago

您可以仔细阅读一下YOLO和DEYO后处理的代码，DEYO是端到端检测器，无需使用NMS进行后处理，DEYO的验证结果与不同的iou无关，同时在验证时您无需调整conf，我们的代码也不支持这个功能。不过您可以通过model.predict()来调整conf，获取理想的预测。

Letitialky commented 4 weeks ago

感谢作者的回复，那请问在论文中与yolov5和yolov8进行比较时，这两个模型在验证时设置的conf和iou为多少呢？

ouyanghaodong commented 3 weeks ago

使用官方默认的conf = 0.001, iou = 0.7