Faster R-CNN系列(二):实现anchor
- 导入相关的包
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.patches as patches
%matplotlib inline
- 定义feature map 的size
- 在这里定义为16*16
- 每个feature map上的点对应原图中8*8的区域
size_X=16 #feature map size 16*16
size_Y=16
rpn_stride=8 #feature 一个点对应原图中的点的个数 原始图片大小 (16*8)*(16*8)
scales=[2,4,8] #尺寸
ratios=[0.5,1,2] #比例
- 展示meshgrid 用法
# meshgrid的用法 将坐标向量转换成坐标矩阵的函数
# f_x=np.arange(size_X)
# f_y=np.arange(size_Y)
# F_X,F_Y=np.meshgrid(f_x,f_y)
# print(F_X)
# print(F_Y)
- 产生anchor
- 输入feature map的 长,宽,采样数,anchor的尺寸和比例
def anchor_gen(size_X,size