U
|
x¨_ü ã @ s> d Z ddlZddlm mZ ddlmZ G dd deZ dS )z9
|
@author: l1aoxingyu
|
@contact: sherlockliao01@gmail.com
|
é N)Úget_event_storagec @ s. e Zd ZdZdd Zed
|
ddZdd Zd S ) ÚCrossEntropyLossz^
|
A class that stores information and compute losses about outputs of a Baseline head.
|
c C s: |j jj| _|j jjj| _|j jjj| _ |j jjj
|
| _ d S )N) ÚMODELÚHEADSÚ NUM_CLASSESÚ _num_classesÚLOSSESÚCEÚEPSILONÚ_epsÚALPHAÚ_alphaÚSCALEÚ_scale)ÚselfÚcfg© r ú&./modeling/losses/cross_entroy_loss.pyÚ__init__ s zCrossEntropyLoss.__init__©é c C s¢ | d¡}t|}| |ddd¡\}}| ¡ }| | dd¡ |¡¡}g }|D ]8} |d|
d¡ ¡ jddd}
|
| |
|
|
d| ¡¡ qNt } | d|d ¡ dS ) z;
|
Log the accuracy metrics to EventStorage.
|
r r TéÿÿÿÿN)ÚdimÚkeepdimg ð?Z cls_accuracy) ÚsizeÚmaxÚtopkÚtÚeqÚviewÚ expand_asÚfloatÚsumÚappendÚmul_r Z
|
put_scalar) Úpred_class_logitsÚ
|
gt_classesr ÚbszZmaxkÚ_Z
|
pred_classÚcorrectÚretÚkZ correct_kÚstorager r r Ú log_accuracy s
|
z CrossEntropyLoss.log_accuracyc C sº | j dkr| j }n2tj|dd}| j|t | d¡¡|f d¡ }tj|dd}t ¡ < t
|
|¡}||| j d 9 }| d|j d¡d| ¡ W 5 Q R X | | d¡ ¡ }|| j S )z{
|
Compute the softmax cross entropy loss for box classification.
|
Returns:
|
scalar Tensor
|
r r )r )r ÚFÚsoftmaxr ÚtorchÚaranger Ú unsqueezeÚ log_softmaxÚno_gradÚ ones_liker Úscatter_ÚdataÚmeanr" r )r r% r& Z smooth_paramZ
|
soft_labelÚ log_probsÚtargetsÚlossr r r Ú__call__* s
|
$
|
|
$zCrossEntropyLoss.__call__N)r )Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r Ú staticmethodr- r< r r r r r s
|
r )
|
r@ r0 Útorch.nn.functionalÚnnÚ
|
functionalr. Z utils.eventsr Úobjectr r r r r Ú<module> s
|
