[cf6a9e]: / loss / __pycache__ / IoU.cpython-35.pyc

Download this file

11 lines (11 with data), 895 Bytes



”¡£_ã@s5ddlZddlTddlZddd„ZdS)éNé)Ú*gà?cCs|dk}|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à?Údimégíµ ÷ư>rr)Ú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_losss 


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