在人脸检测任务下,输入分辨率为 224,数据集为 COCO,stride 分别为 32、16、8 时,计算 YOLOv5 anchors 的宽高如下:

  1. 首先,根据 COCO 数据集中的人脸大小分布,可以得到一些参考值作为初始 anchor 的宽高,比如:
  • 小人脸:宽高为 32x32
  • 中等人脸:宽高为 64x64
  • 大人脸:宽高为 128x128
  1. 然后,根据 stride 的不同,可以计算出在不同特征图上的 anchor 的宽高。假设在某个特征图上,特征图大小为 SxS,anchor 的比例为 p(即宽高比),则该特征图上的 anchor 的宽高可以计算为:
  • 宽:w = p * sqrt(A),其中 A 为 anchor 的面积,可以根据参考值计算得到,比如 A=32x32、64x64、128x128 等
  • 高:h = p * sqrt(A)
  1. 最后,根据不同的 stride 和特征图大小,可以得到不同大小的 anchor。比如:
  • stride=32,特征图大小为 7x7,可以得到 3 个大小的 anchor,分别为 32x32、64x64、128x128
  • stride=16,特征图大小为 14x14,可以得到 3 个大小的 anchor,分别为 64x64、128x128、256x256
  • stride=8,特征图大小为 28x28,可以得到 3 个大小的 anchor,分别为 128x128、256x256、512x512

需要注意的是,这里的宽高计算方式只是一种常见的方式,具体的计算方式可能因任务、数据集、模型等因素而有所不同。


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

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