Datasets
Models
Downloads: 1
50 lines (50 with data), 6.5 kB
B
%YfÂ& ã @ s d Z ddlZddlmZ ddlmZ ddlmZ dddZG dd d ej Z
G d
d d ej Z G d d
d
ej Z G dd dZ
dS )z
Loss functions
é N)Úbbox_iou)Ú de_parallelç¹?c C s dd| d| fS )Ng ð?g à?© )Úepsr r úI/home/iml/Desktop/Talha/yolov5_P2_ROI_HCM_Micro_1000_Github/utils/loss.pyÚ
smooth_BCE
s r c s& e Zd Zd fdd Zdd Z ZS )ÚBCEBlurWithLogitsLossç©?c s" t ¡ tjdd| _|| _d S )NÚnone)Ú reduction)ÚsuperÚ__init__ÚnnÚBCEWithLogitsLossÚloss_fcnÚalpha)Úselfr )Ú __class__r r r s
z BCEBlurWithLogitsLoss.__init__c C sJ | ||¡}t |¡}|| }dt |d | jd ¡ }||9 }| ¡ S )Né g-C ëâ6?)r ÚtorchÚsigmoidÚexpr Úmean)r ÚpredÚtrueÚlossÚdxÚ alpha_factorr r r Úforward s
z BCEBlurWithLogitsLoss.forward)r
)Ú__name__Ú
__module__Ú __qualname__r r Ú
__classcell__r r )r r r s r c s& e Zd Zd fdd Zdd Z ZS )Ú FocalLossç ø?ç Ð?c s0 t ¡ || _|| _|| _|j| _d| j_d S )Nr )r
r r Úgammar r )r r r' r )r r r r % s
zFocalLoss.__init__c C s | ||¡}t |¡}|| d| d| }|| j d| d| j }d| | j }||| 9 }| jdkrv| ¡ S | jdkr| ¡ S |S d S )Nr g ð?r Úsum)r r r r r' r r r( )r r r r Ú pred_probÚp_tr Úmodulating_factorr r r r - s
zFocalLoss.forward)r% r& )r r! r" r r r# r r )r r r$ # s r$ c s& e Zd Zd fdd Zdd Z ZS )Ú
QFocalLossç ø?ç Ð?c s0 t ¡ || _|| _|| _|j| _d| j_d S )Nr )r
r r r' r r )r r r' r )r r r r C s
zQFocalLoss.__init__c C s~ | ||¡}t |¡}|| j d| d| j }t || ¡| j }||| 9 }| jdkrd| ¡ S | jdkrv| ¡ S |S d S )Nr r r( ) r r r r Úabsr' r r r( )r r r r r) r r+ r r r r K s
zQFocalLoss.forward)r- r. )r r! r" r r r# r r )r r r, A s r, c @ s* e Zd ZdZd ddZdd Zdd ZdS )
Ú ComputeLossFc C s$ t | ¡ j}|j}tjtj|d g|dd}tjtj|d g|dd}t| dd¡d\| _
| _ |d }|d krt ||t || }}t
|jd
}d d d
dgi |jd d
dddg¡| _|rÒt|j d¡nd | _||d
||f\| _| _| _| _| _|j| _|j| _|j| _|j| _|| _d S )NÚcls_pw)Údevice)Ú
pos_weightÚobj_pwÚlabel_smoothingg )r Úfl_gammar éÿÿÿÿé g @g ð?gÙ?g Ð?g¸
ëQ¸®?g{®Gáz?é ) ÚnextÚ
parametersr2 Úhypr r r Útensorr ÚgetÚcpÚcnr$ r ÚmodelÚnlÚbalanceÚlistÚstrideÚindexÚssiÚBCEclsÚBCEobjÚgrÚ autobalanceÚnaÚncÚanchors) r rA rK r2 ÚhrH rI ÚgÚmr r r r _ s" $"zComputeLoss.__init__c s® t jd jd}t jd jd}t jd jd} ||¡\}}}} xðt|D ]â\}
} ||
\} }
}}t j| jd d
| j jd}| jd }|rè| | |
||f ddd jfd¡\}}}}| ¡ d d }| ¡ d d | |
}t
||fd¡}t |||
dd ¡ }|d
|
¡ 7 }| ¡ d¡ |j¡} jrr| ¡ }| | |
| || || || f\} }
}}} jdk rd
j j| }||| |
||f< jdkrèt j| j jd} j|t|||
f< | ||¡7 } | d |¡}|| j|
7 } jrP j|
d d
| ¡ ¡ j|
< qPW jrV fdd jD _| j d 9 }| j d 9 }| j d 9 }|jd }|| | | t
|||f¡ ¡ fS )Nr )r2 é )Údtyper2 r é g à?T)ÚCIoUg ð?).rR g §èH.ÿï?g-C ëâ6?c s g | ]}| j j qS r )rC rG )Ú.0Úx)r r r ú
<listcomp>© s z(ComputeLoss.__call__.<locals>.<listcomp>ÚboxÚobjÚcls) r Úzerosr2 Ú
build_targetsÚ enumerateÚshaperS ÚsplitrM r Úcatr Úsqueezer ÚdetachÚclampÚtypeÚ sort_obj_iouÚargsortrJ Ú full_liker@ r? ÚrangerH rI rC rK Úitemr< ) r ÚpÚtargetsZlclsZlboxÚlobjÚtclsÚtboxÚindicesrN ÚiÚpiÚbÚaÚgjÚgiZtobjÚnÚpxyZpwhÚ_ZpclsZpboxÚiouÚjÚtZobjiÚbsr )r r Ú__call__y sJ
*, (
zComputeLoss.__call__c ! C sÀ | j |jd }}g g g g f\}}}}tjd| jd} tj|| jd ¡ |d¡ d|¡}
t | |dd¡|
d fd¡}d} tj
ddgddgddgddgddgg| jd ¡ | } xþt | j D ]î}
| j
|
||
j }}t
|¡d dd dg | dd
< || }|rø|d d d
f |d d
d f }t |d| ¡ d¡d | jd
k }|| }|d d
dd
f }| dd g | }|d | k |dk@ j\}}|d | k |dk@ j\}}t t |¡||||f¡}| d¡| }t |¡d | d d
d f | }n |d }d}| d d¡\}}}}| ¡ d¡| ¡ j }\} } || ¡ } | j\} } | | || d|d d ¡| d|d d ¡f¡ | t || |fd¡¡ | || ¡ | | ¡ qÀW ||||fS )Nr é )r2 r ).NrT g à?r7 r8 é .rR Úanchor_t)é r r )rL r_ r Úonesr2 ÚarangeÚfloatÚviewÚrepeatra r= ri rB rN Úmaxr< ÚTÚstackÚ ones_likeÚ
zeros_likeÚchunkÚlongÚappendÚclamp_)!r rk rl rL Úntrn ro rp ZanchÚgainÚairP Úoffrq rN r_ r| Úrr{ ZgxyZgxiÚkÚlrQ ÚoffsetsÚbcZgwhrt rs ÚcZgijrv ru r r r r] ± sN $ $$
2zComputeLoss.build_targetsN)F)r r! r" rf r r~ r] r r r r r0 [ s
8r0 )r )Ú__doc__r Útorch.nnr Ú
utils.metricsr Úutils.torch_utilsr r ÚModuler r$ r, r0 r r r r Ú<module> s