Datasets
Models
Downloads: 1
19 lines (19 with data), 2.0 kB
”¡£_° ã @ s~ d d l Z d d l Td d l m Z d d l j j Z d d l m Z Gd d „ d e j ƒ Z Gd d „ d e j ƒ Z
d S)
é Né )Ú*)Únn)ÚVariablec s: e Z d Z d Z d d ‡ f d d † Z d d „ Z ‡ S)Ú FocalLossz.
Focal loss for binary classification
é g Ð?c s) t t | ƒ j ƒ | | _ | | _ d S)N)Úsuperr Ú__init__ÚgammaÚalpha)Úselfr
r )Ú __class__© úH/raid/data/wangqiuli/Documents/YH_Interpretable/Version1/losses/focal.pyr s zFocalLoss.__init__c C sŽ t j | d d ƒ} | d d … d f } | j d | | j | t j | ƒ d | j | | j d | t j d | ƒ } | j ƒ S)NÚdimr )ÚtorchÚsoftmaxr r
ÚlogÚmean)r ÚinputÚtargetÚptÚpÚlossr r r Úforward s
'2zFocalLoss.forward)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r r r )r
r r s r c s: e Z d Z d d d d ‡ f d d † Z d d „ Z ‡ S) Ú FocalLoss2dr r Téÿ c s; t t | ƒ j ƒ | | _ | | _ | | _ | | _ d S)N)r r r r r
Ú ignore_indexÚ size_average)r r r
r" r! )r
r r r s
zFocalLoss2d.__init__c C s† t | ƒ } t | ƒ } t j | | d d d | j ƒ} t j | ƒ } | j d | | j | } | j rx | j
ƒ S| j ƒ Sd S)NÚ reductionÚnoner! r ) Úto_float_and_cudaÚto_long_and_cudaÚFÚ
cross_entropyr! r Úexpr r
r" r Úsum)r ÚinputsÚtargetsZce_lossr Z
focal_lossr r r r # s !
zFocalLoss2d.forward)r r r r r r r )r
r r s r ) r Úutilsr Útorch.nn.functionalÚ
functionalr' Útorch.autogradr ÚModuler r r r r r Ú<module> s