retinanet之anchor

最新推荐文章于 2023-04-05 15:51:35 发布

烽火四起，烟雨平生

最新推荐文章于 2023-04-05 15:51:35 发布

阅读量1.4k

点赞数

文章标签：机器学习

本文链接：https://blog.csdn.net/qq_43724644/article/details/105564386

版权

retinanet之anchor

1.综述

anchor即目标检测的先验框，retinanet的anchor有scales和ratios，根据每个pyramid_levels生成三个scales，三个ratios，然后形成九种先验框。
scales：anchor的大小（宽）
ratios：anchor的宽高比

2.参数：pyramid_levels, strides, sizes, ratios, scales

class Anchors(nn.Module):
    def __init__(self, pyramid_levels=None, strides=None, sizes=None, ratios=None, scales=None):
        super(Anchors, self).__init__()

        if pyramid_levels is None:
            self.pyramid_levels = [3, 4, 5, 6, 7]
        if strides is None:
            self.strides = [2 ** x for x in self.pyramid_levels]
        if sizes is None:
            self.sizes = [2 ** (x + 2) for x in self.pyramid_levels]
        if ratios is None:
            self.ratios = np.array([0.5, 1, 2])
        if scales is None:
            self.scales = np.array([2 ** 0, 2 ** (1.0 / 3.0), 2 ** (2.0 / 3.0)])