From 37c4ff8d0ba776c17461709e2083d838404dfbe0 Mon Sep 17 00:00:00 2001 From: np-csu Date: Mon, 1 Oct 2018 18:10:33 +0800 Subject: [PATCH] Update proposal_target.py --- example/rcnn/symnet/proposal_target.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/rcnn/symnet/proposal_target.py b/example/rcnn/symnet/proposal_target.py index 926720b28b5f..77cb78850176 100644 --- a/example/rcnn/symnet/proposal_target.py +++ b/example/rcnn/symnet/proposal_target.py @@ -35,7 +35,7 @@ def sample_rois(rois, gt_boxes, num_classes, rois_per_image, fg_rois_per_image, :param fg_rois_per_image: foreground roi number :param fg_overlap: overlap threshold for fg rois :param box_stds: std var of bbox reg - :return: (labels, rois, bbox_targets, bbox_weights) + :return: (rois, labels, bbox_targets, bbox_weights) """ overlaps = bbox_overlaps(rois[:, 1:], gt_boxes[:, :4]) gt_assignment = overlaps.argmax(axis=1)