399a74dfa594ec891c1da5553394b8155f64453d,examples/FasterRCNN/modeling/model_cascade.py,CascadeRCNNHead,match_box_with_gt,#CascadeRCNNHead#Any#Any#,79

Before Change

            with tf.name_scope("match_box_with_gt_{}".format(iou_threshold)):
                iou = pairwise_iou(boxes, self.gt_boxes)  // NxM
                max_iou_per_box = tf.reduce_max(iou, axis=1)  // N
                best_iou_ind = tf.argmax(iou, axis=1)  // N
                labels_per_box = tf.gather(self.gt_labels, best_iou_ind)
                fg_mask = max_iou_per_box >= iou_threshold
                fg_inds_wrt_gt = tf.boolean_mask(best_iou_ind, fg_mask)

After Change

            with tf.name_scope("match_box_with_gt_{}".format(iou_threshold)):
                iou = pairwise_iou(boxes, self.gt_boxes)  // NxM
                max_iou_per_box = tf.reduce_max(iou, axis=1)  // N
                best_iou_ind = tf.cond(tf.shape(iou)[1] > 0,
                                       lambda: tf.argmax(iou, axis=1),   // //proposal, each in 0~m-1
                                       lambda: tf.zeros([tf.shape(iou)[0]], dtype=tf.int64))
                labels_per_box = tf.gather(self.gt_labels, best_iou_ind)
                fg_mask = max_iou_per_box >= iou_threshold

Instances