bubbliiiing / yolov5-v6.1-pytorch

这是一个yolov5-v6.1-pytorch的源码,可以用于训练自己的模型。

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

请问可以计算测试集上的iou吗?类似get_map的功能

Sudae827 opened this issue · comments

commented

额0 0iou?iou是针对每个目标的呀

commented

目前测试集也有对应的标签。有个想法是求解测试集上每个目标预测框和真实框的iou,然后取平均值获得位置准确性的评价指标(看到一一些论文用map衡量分类准确度,用iou衡量定位准确度)。但目前存在问题是:预测框的数量和真实框数量不一致,无法做到一一对应。
非科班出身的入门级选手,可能理解存在一定问题。还请b导能指点一下上述思路是否可行?能否把这个指标也加入评价系统中。

额0 0,我感觉直接看mAP50,mAP75这种就可以欸

想請問為什麼我使用predict.py出來的結果是正常的
但使用get_map.py出來的結果有很多亂框?是有哪裡需要更改嗎?