11 lines (11 with data), 895 Bytes
¡£_ ã @ s5 d d l Z d d l Td d l Z d d d Z d S)é Né )Ú*g à?c C s | d k } | t j | k } t | } t | } | | } t j | d d } t j | d d } | | } t j | d d } t j | d d } | | } t j | d d } t j | d d } | | | | d } d d t j | | j d } | S)zw
2d dice loss
:param input: predict tensor
:param target: target tensor
:return: scalar loss value
g à?Údimé gíµ ÷ư>r r )ÚtorchÚmaxÚto_float_and_cudaÚsumÚsize)ÚinputÚtargetÚ thresholdÚnumÚden1Úden2ÚiouZ iou_total© r úF/raid/data/wangqiuli/Documents/YH_Interpretable/Version1/losses/IoU.pyÚIoU_loss s
$r )r ÚutilsÚnumpyÚnpr r r r r Ú<module> s