YOLOv5 Anchor 尺寸计算:人脸检测任务下输入分辨率为224的案例
在人脸检测任务下,输入分辨率为 224,数据集为 COCO,stride 分别为 32、16、8 时,计算 YOLOv5 anchors 的宽高如下:
- 首先,根据 COCO 数据集中的人脸大小分布,可以得到一些参考值作为初始 anchor 的宽高,比如:
- 小人脸:宽高为 32x32
- 中等人脸:宽高为 64x64
- 大人脸:宽高为 128x128
- 然后,根据 stride 的不同,可以计算出在不同特征图上的 anchor 的宽高。假设在某个特征图上,特征图大小为 SxS,anchor 的比例为 p(即宽高比),则该特征图上的 anchor 的宽高可以计算为:
- 宽:w = p * sqrt(A),其中 A 为 anchor 的面积,可以根据参考值计算得到,比如 A=32x32、64x64、128x128 等
- 高:h = p * sqrt(A)
- 最后,根据不同的 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 著作权归作者所有。请勿转载和采集!