RPN.py 文件源码

def proposal_layer(rpn_cls_prob_reshape, rpn_bbox_pred, im_info, cfg_key, 
                    _feat_stride, anchor_scales, anchor_ratios, is_region):
        rpn_cls_prob_reshape = rpn_cls_prob_reshape.data.cpu().numpy()
        rpn_bbox_pred = rpn_bbox_pred.data.cpu().numpy()
        x = proposal_layer_py(rpn_cls_prob_reshape, rpn_bbox_pred, im_info, 
                    cfg_key, _feat_stride, anchor_scales, anchor_ratios, is_region=is_region)
        x = network.np_to_variable(x, is_cuda=True)
        return x.view(-1, 5)