在人脸检测任务下,输入分辨率为 224,数据集为 COCO,可以使用以下公式计算 YOLOv5 Anchors 的宽高:

  1. 首先,根据 COCO 数据集中所有人脸的宽度和高度,计算出宽度和高度的平均值:
avg_width = sum(所有人脸的宽度) / 总人脸数
avg_height = sum(所有人脸的高度) / 总人脸数
  1. 接下来,根据输入分辨率和 YOLOv5 的 Anchor 比例,计算出相应的 Anchor 大小:
anchor_width = avg_width / 输入分辨率 * anchor 比例
anchor_height = avg_height / 输入分辨率 * anchor 比例

其中,Anchor 比例是 YOLOv5 中预定义的一组 Anchor 大小,通常有 3 个或 5 个,可以根据实际情况选择。

例如,如果选择 3 个 Anchor 比例,分别为 [1.0, 1.5, 2.0],则可以计算出 3 个 Anchor 的宽高:

anchor1_width = avg_width / 输入分辨率 * 1.0
anchor1_height = avg_height / 输入分辨率 * 1.0

anchor2_width = avg_width / 输入分辨率 * 1.5
anchor2_height = avg_height / 输入分辨率 * 1.5

anchor3_width = avg_width / 输入分辨率 * 2.0
anchor3_height = avg_height / 输入分辨率 * 2.0

这样就可以得到 YOLOv5 在人脸检测任务下的 Anchor 大小了。

YOLOv5 人脸检测 Anchor 计算 - 输入分辨率 224,COCO 数据集

原文地址: https://www.cveoy.top/t/topic/jwPA 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录