RoIHead:负责对rois进行分类和回归微调。
所以Faster RCNN的流程可以总结为:
原始图像--->特征提取------>RPN产生候选框------>对候选框进行分类和回归微调。...# 去正则化,img维度为[[B,G,R],H,W],因为caffe预训练模型输入为BGR 0-255图片,pytorch预训练模型采用RGB 0-1图片
def inverse_normalize(img...-255
return (img * 0.225 + 0.45).clip(min=0, max=1) * 255
# 采用pytorch预训练模型对图片预处理,函数输入的img为0-1
def...# 对相应的bounding boxes 也也进行同等尺度的缩放。..., W * scale), mode='reflect',anti_aliasing=False)
#调用pytorch_normalze或者caffe_normalze对图像进行正则化