7f2a17f685b1088b50fc1f522e180f10502b6db2,dl/callbacks.py,IOUCallback,on_batch_end,#IOUCallback#Any#,167

Before Change

        // msk_vpreds = msk_vpreds.sigmoid()
        msk_vpreds = msk_vpreds.detach().cpu().numpy()
        valid_msks = valid_msks.detach().cpu().numpy()

        iou = get_iou_vector(valid_msks, msk_vpreds)
        key = "iou"
        state.batch_metrics[key] = iou

After Change

        outputs = state.output[self.output_key]
        targets = state.input[self.input_key]

        jac = jaccard(targets, (outputs > 0).float())
        key = "jaccard"
        state.batch_metrics[key] = jac

Instances