faster_rcnn.py 文件源码

python
阅读 25 收藏 0 点赞 0 评论 0

项目:pytorch_RFCN 作者: PureDiors 项目源码 文件源码
def proposal_layer(rpn_cls_prob_reshape, rpn_bbox_pred, im_info, cfg_key, _feat_stride, anchor_scales):
        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)
        x = network.np_to_variable(x, is_cuda=True)
        return x.view(-1, 5)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号