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
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
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
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
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
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