2d44d66f74c0be1f2952af80cc11a9076fd71e3c,chainercv/links/model/mask_rcnn/mask_head.py,,mask_loss_post,#Any#Any#Any#Any#Any#,275

Before Change

    mask_loss = 0
    for i in np.unique(cuda.to_cpu(mask_roi_indices)):
        index = (mask_roi_indices == i).nonzero()[0]
        gt_segm = gt_segms[index]
        gt_mask_label = gt_mask_labels[index]

        mask_loss += F.sigmoid_cross_entropy(
            segms[index, gt_mask_label], gt_segm.astype(np.int32))

After Change

    gt_mask_labels = xp.hstack(gt_mask_labels).astype(np.int32)

    mask_loss = F.sigmoid_cross_entropy(
        segms[np.arange(len(gt_mask_labels)), gt_mask_labels],
        gt_segms.astype(np.int32))
    return mask_loss

Instances