Python proposal_top_layerの例

コード例 #1

ファイル: network_.py プロジェクト: ZENGXH/pytorch-faster-rcnn

 def _proposal_top_layer(self, rpn_cls_prob, rpn_bbox_pred):
     rois, rpn_scores = proposal_top_layer(\
                                     rpn_cls_prob.data.cpu().numpy(), rpn_bbox_pred.data.cpu().numpy(), self._im_info,
                                      self._feat_stride, self._anchors, self._num_anchors)
     rois_, rpn_scores_ = proposal_top_layer_(\
                                     rpn_cls_prob, rpn_bbox_pred, self._im_info,
                                      self._feat_stride, torch.from_numpy(self._anchors), self._num_anchors)
     return rois, rpn_scores

コード例 #2

ファイル: network.py プロジェクト: rnsandeep/tf-faster-rcnn-visualize

    def _proposal_top_layer(self, rpn_cls_prob, rpn_bbox_pred, name):
        with tf.variable_scope(name) as scope:
            rois, rpn_scores = proposal_top_layer(rpn_cls_prob, rpn_bbox_pred,
                                                  self._im_info,
                                                  self._feat_stride,
                                                  self._anchors,
                                                  self._num_anchors)
            rois.set_shape([cfg.TEST.RPN_TOP_N, 5])
            rpn_scores.set_shape([cfg.TEST.RPN_TOP_N, 1])

        return rois, rpn_scores

コード例 #3

ファイル: network.py プロジェクト: guanlongzhao/dehaze

  def _proposal_top_layer(self, rpn_cls_prob, rpn_bbox_pred, name):
    with tf.variable_scope(name) as scope:
      rois, rpn_scores = proposal_top_layer(
        rpn_cls_prob,
        rpn_bbox_pred,
        self._im_info,
        self._feat_stride,
        self._anchors,
        self._num_anchors
      )
      rois.set_shape([cfg.TEST.RPN_TOP_N, 5])
      rpn_scores.set_shape([cfg.TEST.RPN_TOP_N, 1])

    return rois, rpn_scores

コード例 #4

ファイル: network_vgg.py プロジェクト: zeyuxiao1997/lang2seg

 def _proposal_top_layer(self, rpn_cls_prob, rpn_bbox_pred):
   rois, rpn_scores = proposal_top_layer(\
                                   rpn_cls_prob, rpn_bbox_pred, self._im_info,
                                    self._feat_stride, self._anchors, self._num_anchors)
   return rois, rpn_scores

コード例 #5

ファイル: network.py プロジェクト: vikash0837/common-space-domain-adaptation

 def _proposal_top_layer_fpn(self, rpn_cls_prob, rpn_bbox_pred, idx):
   rois, rpn_scores = proposal_top_layer(\
                                   rpn_cls_prob, rpn_bbox_pred, self._im_info,
                                    [self._feat_stride[idx]], self._anchors[idx], self._num_anchors)
   return rois, rpn_scores

コード例 #6

ファイル: network.py プロジェクト: sunshinezhihuo/AlphaPose

 def _proposal_top_layer(self, rpn_cls_prob, rpn_bbox_pred):
   rois, rpn_scores = proposal_top_layer(\
                                   rpn_cls_prob, rpn_bbox_pred, self._im_info,
                                    self._feat_stride, self._anchors, self._num_anchors)
   return rois, rpn_scores