import numpy as np

import torch

def meandice(pred, gt, dices):

    """

    :return save img' dice value in IoUs

    """

    # dices = []

    pred[pred < 0.5] = 0

    pred[pred >= 0.5] = 1

    gt[gt < 0.5] = 0

    gt[gt >= 0.5] = 1

    pred = pred.type(torch.LongTensor)

    pred_np = pred.data.cpu().numpy()

    gt = gt.data.cpu().numpy()

    for x in range(pred.size()[0]):

        dice = np.sum(pred_np[x][gt[x] == 1]) * 2 / float(np.sum(pred_np[x]) + np.sum(gt[x]))

        dices.append(dice)

    return dices