From 4ac0f943224e138fede6b23b6b80181e5a015860 Mon Sep 17 00:00:00 2001 From: Ahmed Shariff Date: Thu, 30 Aug 2018 14:22:43 +0530 Subject: [PATCH] Replaced the use of Tensor.index(m) with Tensor[m] --- lib/model/rpn/proposal_target_layer_cascade.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/model/rpn/proposal_target_layer_cascade.py b/lib/model/rpn/proposal_target_layer_cascade.py index 249439964..ddcccad3f 100644 --- a/lib/model/rpn/proposal_target_layer_cascade.py +++ b/lib/model/rpn/proposal_target_layer_cascade.py @@ -130,7 +130,7 @@ def _sample_rois_pytorch(self, all_rois, gt_boxes, fg_rois_per_image, rois_per_i offset = torch.arange(0, batch_size)*gt_boxes.size(1) offset = offset.view(-1, 1).type_as(gt_assignment) + gt_assignment - labels = gt_boxes[:,:,4].contiguous().view(-1).index((offset.view(-1),)).view(batch_size, -1) + labels = gt_boxes[:,:,4].contiguous().view(-1)[(offset.view(-1),)].view(batch_size, -1) labels_batch = labels.new(batch_size, rois_per_image).zero_() rois_batch = all_rois.new(batch_size, rois_per_image, 5).zero_()