test_anchor.py

import os,sys,json
from pathlib import Path
import tensorflow as tf
PATH =Path(os.getcwd()).parent
sys.path.append(str(PATH.absolute()) + "/")
PATH =Path(os.getcwd())
sys.path.append(str(PATH.absolute()) + "/")
from mmdet import core

config_anchor_generator = dict(
    type='AnchorGenerator',
    strides=[10,],
    ratios=[1.,],
    scales=[1.,],
    base_sizes=[10],
)

anchor_generator = core.build_anchor_generator(config_anchor_generator)
all_anchors = anchor_generator.grid_anchors([(2, 2)])
expected_anchors = tf.convert_to_tensor([[-5., -5., 5., 5.], [5., -5., 15., 5.],
                                     [-5., 5., 5., 15.], [5., 5., 15., 15.]])

assert tf.math.reduce_mean(tf.math.abs(all_anchors[0] - expected_anchors)) < 1e-5


config_anchor_generator = dict(
    type='AnchorGenerator',
    strides=[(10,20)],
    ratios=[1.,],
    scales=[1.,],
    base_sizes=[10],
)

anchor_generator = core.build_anchor_generator(config_anchor_generator)
all_anchors = anchor_generator.grid_anchors([(2, 2)])
expected_anchors = tf.convert_to_tensor([[-5., -5., 5., 5.], [5., -5., 15., 5.],
                                     [-5., 15., 5., 25.], [5., 15., 15., 25.]])

assert tf.math.reduce_mean(tf.math.abs(all_anchors[0] - expected_anchors)) < 1e-5

print("pass all")