在 OpenVINO™ 精度检查器工具中添加 DIoU-NMS 指标,以获取 YOLO v4 的正确 mAP
openlab_4276841a
更新于 3年前
Yolov4 模型于 2020 年中期推出,它的问世对深度学习目标检测领域产生了深远影响。为取得成功,Yolov4 集成了许多一流的技术,其中包括改进的非极大值抑制 (NMS) 算法,该算法使用距离交并比 (DIoU) 而非交并比 (IoU)。DIoU 总结了边界框回归中的两个几何因素(即重叠区域和中心点距离),从而加快了收敛速度并提升了性能。
在本***中,DIoU-NMS 函数将被添加到 OpenVINO™ 精度检查器工具中,以用于计算yolov4 的正确预期精度。尽管 DIoU-NMS 可帮助大幅提升精度,但它也可以与训练后优化工具包 (pot) 一起使用,以生成优化的 INT8 模型。
0个评论