cd48957b671f2799c19578f73025da50dd100de5,lib/model/rpn/rpn.py,_RPN,forward,#_RPN#,58

Before Change

            fg_cnt = 0
            self.rpn_loss_cls = 0
            for i in range(batch_size):
                rpn_keep = rpn_label[i].ne(-1).nonzero().squeeze()
                rpn_keep_v = Variable(rpn_keep)
                rpn_cls_score_single = torch.index_select(rpn_cls_score[i], 0, rpn_keep_v)
                rpn_label_tmp = torch.index_select(rpn_label[i], 0, rpn_keep)

After Change

            // self.rpn_loss_cls /= batch_size

            rpn_keep = Variable(rpn_label.view(-1).ne(-1).nonzero().view(-1))
            rpn_cls_score = torch.index_select(rpn_cls_score.view(-1,2), 0, rpn_keep)
            rpn_label = torch.index_select(rpn_label.view(-1), 0, rpn_keep.data)
            rpn_label = Variable(rpn_label.long())

Instances