8dd1ec5bfef42af8ad75a38a4ac6d58e7d7763c9,luminoth/utils/image_vis.py,,draw_rpn_bbox_pred,#Any#Any#,598

Before Change

    
    logger.debug("RPN regression loss (bbox to original anchors, with the smoothL1Loss)")
    target = pred_dict["rpn_prediction"]["rpn_cls_target"]
    target = target.reshape([-1, 1])
    // Get anchors with positive label.
    positive_indices = np.nonzero(np.squeeze(target) > 0)[0]
    random_indices = np.random.choice(np.arange(len(positive_indices)), n)

    loss_per_anchor = pred_dict["rpn_prediction"]["reg_loss_per_anchor"]

    // Get only n random to avoid overloading image.
    positive_indices = positive_indices[random_indices]
    loss_per_anchor = loss_per_anchor[random_indices]
    target = target[positive_indices]

    bbox_pred = pred_dict["rpn_prediction"]["rpn_bbox_pred"]
    bbox_pred = bbox_pred.reshape([-1, 4])
    bbox_pred = bbox_pred[positive_indices]

After Change

    loss_per_anchor = pred_dict["rpn_prediction"]["reg_loss_per_anchor"]

    top_losses_idx = np.argsort(loss_per_anchor)[::-1][:top_k]

    loss_per_anchor = loss_per_anchor[top_losses_idx]
    bbox_pred = bbox_pred[top_losses_idx]
    all_anchors = all_anchors[top_losses_idx]

    bbox_final = decode(all_anchors, bbox_pred)

    image_pil, draw = get_image_draw(pred_dict)

Instances