7 lines (7 with data), 780 Bytes

3

 \–aã@s&dd
lZddl
Tdd
lZddd„
Zd
S)éNé
)
Ú
*çà?cCsÆ|d
k}|
tj
|
ƒ
k}
t|ƒ
}t|
ƒ
}
||
}tj|dd}tj|dd}||}tj|dd}tj|dd}|
|
}tj|dd}tj|dd}||||d}ddtj|ƒ
|jdƒ
}|S)zw
    2d dice loss
    :param input: predict tensor
    :param target: target tensor
    :return: scalar loss value
    gà?é)
Údimgíµ ÷Æ°>rr
)ÚtorchÚmaxÚto_float_and_cudaÚsumÚsize)ÚinputÚtargetÚ	thresholdÚnumÚden1Úden2ÚiouZ	iou_total©rú&/home/ws/yanghan/Newcode/losses/IoU.pyÚIoU_losss 










r)
r)rÚutilsÚnumpyÚnprrrrrÚ<module>
s