Comprensión superficial de Anchor y RPN (3) - Análisis de código fuente de generación de Anchor en mmdetection

En MMDetection, la red RPN usa la clase AnchorGenerator para generar Anchor. La configuración predeterminada de AnchorGenerator en el archivo de configuración es la siguiente:

anchor_generator=dict(
    type='AnchorGenerator',
    scales=[8],
    ratios=[0.5,1.0,2.0], 
    strides=[4,8,16,32,64]),

A continuación, llame directamente a la clase AnchorGenerator y su directorio de código fuente de implementación es: mmdet\core\anchor\anchor_generator.py

En esta clase, existen los siguientes métodos más importantes, y existe una comprensión preliminar:

  • Parámetros y llamadas de demostración: si no comprende los parámetros, puede dejar una impresión en su mente, y el siguiente es un análisis detallado. Demo copia directamente el oficial:
    Args:
        strides (list[int] | list[tuple[int, int]]): 各尺度 feature map 相对于原图的缩放因子,也可以叫做anchor移动的步长
        ratios (list[float]): anchor 的高宽比列表
        scales (list[int] | None): anchor 的缩放比率,不能与 `octave_base_scale` 和 `scales_per_octave	

Supongo que te gusta

Origin blog.csdn.net/qq_42308217/article/details/122791815
Recomendado
Clasificación