From 717c0600615a0177fe8d4d82aa4b0f7dfb052186 Mon Sep 17 00:00:00 2001 From: elias-1 Date: Wed, 25 Jan 2017 14:27:15 +0800 Subject: [PATCH] fix index bug.the original index is a float datatype --- lib/rpn_msr/proposal_target_layer_tf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/rpn_msr/proposal_target_layer_tf.py b/lib/rpn_msr/proposal_target_layer_tf.py index a5a3e8e0..03adda55 100644 --- a/lib/rpn_msr/proposal_target_layer_tf.py +++ b/lib/rpn_msr/proposal_target_layer_tf.py @@ -84,7 +84,7 @@ def _get_bbox_regression_labels(bbox_target_data, num_classes): inds = np.where(clss > 0)[0] for ind in inds: cls = clss[ind] - start = 4 * cls + start = int(4 * cls) end = start + 4 bbox_targets[ind, start:end] = bbox_target_data[ind, 1:] bbox_inside_weights[ind, start:end] = cfg.TRAIN.BBOX_INSIDE_WEIGHTS @@ -122,7 +122,7 @@ def _sample_rois(all_rois, gt_boxes, fg_rois_per_image, rois_per_image, num_clas fg_inds = np.where(max_overlaps >= cfg.TRAIN.FG_THRESH)[0] # Guard against the case when an image has fewer than fg_rois_per_image # foreground RoIs - fg_rois_per_this_image = min(fg_rois_per_image, fg_inds.size) + fg_rois_per_this_image = int(min(fg_rois_per_image, fg_inds.size)) # Sample foreground regions without replacement if fg_inds.size > 0: fg_inds = npr.choice(fg_inds, size=fg_rois_per_this_image, replace=False)