Datasets
Models
Downloads: 1
18 lines (18 with data), 1.9 kB
B
`Éa° ã @ sb d dl Z ddlT d dl mZ d dlm mZ d dlmZ G dd dejZ G dd d ejZ
dS )
é Né )Ú*)Únn)ÚVariablec s* e Zd ZdZd fdd Zdd Z ZS ) Ú FocalLossz.
Focal loss for binary classification
é ç Ð?c s t t| ¡ || _|| _d S )N)Úsuperr Ú__init__ÚgammaÚalpha)Úselfr r )Ú __class__© ú3/home/wangqiuli/Code/Newcode1222_monk/loss/focal.pyr
s zFocalLoss.__init__c C st t j|dd}|d d
df }| j d| | j |t |¡ d| j || j d| t d| ¡ }| ¡ S )Nr )Údim)ÚtorchÚsoftmaxr r ÚlogÚmean)r
ÚinputÚtargetÚptÚpÚlossr r r Úforward s
",zFocalLoss.forward)r r )Ú__name__Ú
__module__Ú __qualname__Ú__doc__r
r Ú
__classcell__r r )r r r s r c s& e Zd Zd fdd Zdd Z ZS )
Ú FocalLoss2dr r Téÿ c s* t t| ¡ || _|| _|| _|| _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 sb t |}t|}tj||d| jd}t | ¡}| jd| | j | }| j rV|
¡ S | ¡ S d S )NÚnone)Ú reductionr# r ) Úto_float_and_cudaÚto_long_and_cudaÚFÚ
cross_entropyr# r Úexpr r r$ r Úsum)r
ÚinputsÚtargetsÚce_lossr Ú
focal_lossr r r r # s zFocalLoss2d.forward)r r Tr" )r r r r
r r r r )r r r! s r! ) r Úutilsr Ztorch.nn.functionalÚ
functionalr) Ztorch.autogradr ÚModuler r! r r r r Ú<module> s