Datasets
Models
Downloads: 1
65 lines (65 with data), 13.7 kB
§
¢eÀ& ã óÊ d Z ddlZddlmZ ddlmZ ddlmZ ddZ G d dej ¦ « Z
G d d
ej ¦ « Z G d d ej ¦ « Z G d
d¦ « Z
dS )z
Loss functions
é N)Úbbox_iou)Ú de_parallelç¹?c ó dd| z z
d| z fS )Nç ð?ç à?© ©Úepss ú9/home/iml/Desktop/Talha/YOLOV5_Model/yolov5/utils/loss.pyÚ
smooth_BCEr
s à s?C #IÐ %Ð%ó c ó& e Zd Zd fd Zd Z xZS )ÚBCEBlurWithLogitsLossç©?c ó t ¦ « ¦ « t j d¬¦ « | _ || _ d S )NÚnone)Ú reduction)ÚsuperÚ__init__ÚnnÚBCEWithLogitsLossÚloss_fcnÚalpha)Úselfr Ú __class__s r r z BCEBlurWithLogitsLoss.__init__ s9 ø Ý
×ÒÑÔÐÝÔ,°vÐ>Ñ>Ô>
Ø
r c óÜ | ||¦ « }t j |¦ « }||z
}dt j |dz
| j dz z ¦ « z
}||z }| ¦ « S )Né ç-C ëâ6?)r ÚtorchÚsigmoidÚexpr Úmean)r ÚpredÚtrueÚlossÚdxÚ alpha_factors r Úforwardz BCEBlurWithLogitsLoss.forward si Ø}}T 4Ñ(Ô(Ý}TÑ"Ô"Ø
D[à59 b¨1¡f°´¸dÑ1BÑ%CÑDÔDÑD Ø Ñ Øyy{{Ðr )r ©Ú__name__Ú
__module__Ú __qualname__r r) Ú
__classcell__©r s @r r r sL ø ðð ð ð ð ð ð
ð ð ð ð ð ð r r c ó& e Zd Zd fd Zd Z xZS )Ú FocalLossç ø?ç Ð?c ó¢ t ¦ « ¦ « || _ || _ || _ |j | _ d| j _ d S ©Nr ©r r r Úgammar r ©r r r7 r r s r r zFocalLoss.__init__% óH ø Ý
×ÒÑÔÐØ
Ø
Ø
Ø!Ô+Ø"(
Ô Ð Ð r c óX | ||¦ « }t j |¦ « }||z d|z
d|z
z z }|| j z d|z
d| j z
z z }d|z
| j z }|||z z }| j dk r| ¦ « S | j dk r| ¦ « S |S )Nr r r# Úsum)r r r! r r7 r r# r; )r r$ r% r& Ú pred_probÚp_tr( Úmodulating_factors r r) zFocalLoss.forward- s¾ Ø}}T 4Ñ(Ô(õ
M $Ñ'Ô' ØYÑ ! d¡(¨q°9©}Ñ!=Ñ=ØdjÑ(¨A°©H¸¸T¼Z¹ Ñ+HÑH Ø 3Y¨4¬:Ñ 5ÐØ Ð 0Ñ0Ñ0à >VÒ #Ð #Ø99;;Ð Ø
^uÒ
$Ð
$Ø88::Ð àKr ©r2 r3 r* r/ s @r r1 r1 # sL ø ð)ð )ð )ð )ð )ð )ðð ð ð ð ð ð r r1 c ó& e Zd Zd fd Zd Z xZS )Ú
QFocalLossr2 r3 c ó¢ t ¦ « ¦ « || _ || _ || _ |j | _ d| j _ d S r5 r6 r8 s r r zQFocalLoss.__init__C r9 r c óZ | ||¦ « }t j |¦ « }|| j z d|z
d| j z
z z }t j ||z
¦ « | j z }|||z z }| j dk r| ¦ « S | j dk r| ¦ « S |S )Nr r# r; ) r r r! r Úabsr7 r r# r; )r r$ r% r& r< r( r> s r r) zQFocalLoss.forwardK s« Ø}}T 4Ñ(Ô(åM $Ñ'Ô' ØdjÑ(¨A°©H¸¸T¼Z¹ Ñ+HÑH Ý !I d¨YÑ&6Ñ 7Ô 7¸4¼:Ñ EÐØ Ð 0Ñ0Ñ0à >VÒ #Ð #Ø99;;Ð Ø
^uÒ
$Ð
$Ø88::Ð àKr r? r* r/ s @r rA rA A sL ø ð)ð )ð )ð )ð )ð )ð
ð
ð
ð
ð
ð
ð
r rA c ó& e Zd ZdZddZd Zd ZdS )Ú ComputeLossFc óx t | ¦ « ¦ « j }|j }t j t
j |d g|¬¦ « ¬¦ « }t j t
j |d g|¬¦ « ¬¦ « }t | dd¦ « ¬¦ « \ | _
| _ |d }|d k r t ||¦ « t ||¦ « }}t |¦ « j d
}d g d ¢i |j g d
¢¦ « | _ |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 éÿÿÿÿé )ç @r gÙ?)rQ r r3 g¸
ëQ¸®?g{®Gáz?é r ) ÚnextÚ
parametersrJ Úhypr r r Útensorr
ÚgetÚcpÚcnr1 r ÚmodelÚnlÚbalanceÚlistÚstrideÚindexÚssiÚBCEclsÚBCEobjÚgrÚ autobalanceÚnaÚncÚanchors) r rZ rd rJ Úhra rb ÚgÚms r r zComputeLoss.__init___ s Ýe×&Ò&Ñ(Ô(Ñ)Ô)Ô0Ø Iõ Ô%µ´ ¸qÀ¼{¸mÐTZÐ1[Ñ1[Ô1[Ð\Ñ\Ô\ÝÔ%µ´ ¸qÀ¼{¸mÐTZÐ1[Ñ1[Ô1[Ð\Ñ\Ô\õ &¨!¯%ª%Ð0AÀ3Ñ*GÔ*GÐHÑHÔHÑð
jMØ q55Ý & v¨qÑ 1Ô 1µ9¸VÀQÑ3GÔ3GFFå Ñ Ô Ô $ RÔ (Ø???Ð+×/Ò/°´Ð6RÐ6RÐ6RÑSÔS Ø/:ÐA4>>×'Ò'¨Ñ+Ô+Ð+ÀØHNÐPVÐX[Ð]^Ð`kÐHkÑE T[ $¤'¨4¬8°TÔ5EØ$Ø$Ø$Øy Ø r c ó` t j d j ¬¦ « }t j d j ¬¦ « }t j d j ¬¦ « } ||¦ « \ }}}} t |¦ « D ]¡\ }
} ||
\ } }
}}t j | j d d
| j j ¬¦ « }| j d }|rØ| | |
||f ddd j fd¦ « \ }}}}| ¦ « dz dz
}| ¦ « dz dz | |
z }t j
||fd¦ « }t |||
d¬ ¦ « ¦ « }|d
|z
¦ « z
}| ¦ « d¦ « |j ¦ « } j r?| ¦ « }| | |
| || || || f\ } }
}}} j dk rd
j z
j |z z }||| |
||f< j dk rYt j | j j ¬¦ « } j |t/ |¦ « ||
f< | ||¦ « z
} | d |¦ « }|| j |
z z
} j rB j |
d z d
| ¦ « ¦ « z z j |
< £ j r fd j D ¦ « _ | j d z }| j d z }| j d z }|j d }||z |z |z t j
|||f¦ « ¦ « fS )Nr rI é )ÚdtyperJ r é r T)ÚCIoUr ).rl g §èH.ÿï?r c ó: g | ]}|j j z S r )r\ r` )Ú.0Úxr s r ú
<listcomp>z(ComputeLoss.__call__.<locals>.<listcomp>© s' ø ÐMÐMÐM¸1A ¤ ¨T¬XÔ 6Ñ 6ÐMÐMÐMr ÚboxÚobjÚcls) r ÚzerosrJ Ú
build_targetsÚ enumerateÚshaperm Úsplitrf r! Úcatr Úsqueezer# ÚdetachÚclampÚtypeÚ sort_obj_iouÚargsortrc Ú full_likerY rX Úrangera rb r\ rd ÚitemrU ) r ÚpÚtargetsÚlclsÚlboxÚlobjÚtclsÚtboxÚindicesrg ÚiÚpiÚbÚaÚgjÚgiÚtobjÚnÚpxyÚpwhÚ_ÚpclsÚpboxÚiouÚjÚtÚobjiÚbss ` r Ú__call__zComputeLoss.__call__y s ø Ý{1 T¤[Ð1Ñ1Ô1Ý{1 T¤[Ð1Ñ1Ô1Ý{1 T¤[Ð1Ñ1Ô1Ø'+×'9Ò'9¸!¸WÑ'EÔ'EÑ$dGWõ q\\ð & [ñ & [EArØ" 1:LAq"bÝ;rx¨¨¨|°2´8ÀDÄKÐPÑPÔPDà
AØñ
1à$& q¨!¨R° |Ô$4×$:Ò$:¸A¸qÀ!ÀTÄWÐ;MÈqÑ$QÔ$QÑ!S!Tð kkmm aÑ'¨#Ñ-Ø{{}} qÑ(¨QÑ.°¸´Ñ;Ý y # s ¨QÑ/Ô/Ý t T¨!¤W°4Ð8Ñ8Ô8×@Ò@ÑBÔBØs×(Ò(Ñ*Ô*Ñ*ð jjll×(Ò(¨Ñ+Ô+×0Ò0°´Ñ<Ô<ØÔ$ð IØ
AØ()¨!¬¨a°¬d°B°q´E¸2¸a¼5À#ÀaÄ&Ð(HÑ%Aq"b #Ø7Q;;Ø ¤=¨D¬G°c©MÑ9CØ%(Q2r\Ñ"ð 7Q;;Ý ¨¨d¬g¸d¼kÐJÑJÔJAØ+/¬7AeAhh Q¤Ð'Ñ(ØDKK¨¨aÑ 0Ô 0Ñ0Dð ;;r &z¨4Ñ0Ô0DØ D4<¨?Ñ*Ñ *DØÔ ð
[Ø"&¤,¨q¤/°FÑ":¸VÀdÇkÂkÁmÄm×FXÒFXÑFZÔFZÑ=ZÑ"Z Qùà Ô ð NØMÐMÐMÐMÀÄ ÐMÑMÔMDLØ Ñ Ø Ñ Ø Ñ Ø
Z]àt dÑ" bÑ(%¬)°T¸4ÀÐ4FÑ*GÔ*G×*NÒ*NÑ*PÔ*PÐPÐPr c ó~ | j |j d }}g g g g f\ }}}}t j d| j ¬¦ « } t j || j ¬¦ « ¦ « |d¦ « d|¦ « }
t j | |dd¦ « |
d fd¦ « }d} t j
ddgddgddgddgddgg| j ¬¦ « ¦ « | z } t | j ¦ « D ] }
| j
|
||
j }}t j
|¦ « g d ¢ | dd
< || z }|r |d d d
f |d d
d f z }t j |d|z ¦ « d¦ « d | j d
k }|| }|d d
dd
f }| ddg |z
}|dz | k |dk z j \ }}|dz | k |dk z j \ }}t j t j |¦ « ||||f¦ « }| d¦ « | }t j |¦ « d | d d
d f z | }n
|d }d}| d d¦ « \ }}}}| ¦ « d¦ « | ¦ « j c}\ } } ||z
¦ « } | j \ } } | | || d|d dz
¦ « | d|d dz
¦ « f¦ « | t j || z
|fd¦ « ¦ « | || ¦ « | | ¦ « ¢||||fS )Nr é rI r ).Nrn r rO )rP rn rP rn é .rl Úanchor_trP )é r r )re rz r ÚonesrJ ÚarangeÚfloatÚviewÚrepeatr| rV r r[ rg ÚmaxrU ÚTÚstackÚ ones_likeÚ
zeros_likeÚchunkÚlongÚappendÚclamp_)!r r r re Úntr r r ÚanchÚgainÚairi Úoffr rg rz r Úrr ÚgxyÚgxiÚkÚlrj ÚoffsetsÚbcÚgwhr r ÚcÚgijr r s! r rx zComputeLoss.build_targets± s² à'-¨Ô*BØ$&¨¨B° NÑ!dGTÝz! D¤KÐ0Ñ0Ô0Ý
\" T¤[Ð
1Ñ
1Ô
1×
7Ò
7Ñ
9Ô
9×
>Ò
>¸rÀ1Ñ
EÔ
E×
LÒ
LÈQÐPRÑ
SÔ
SÝ)W^^¨B°°1Ñ 5Ô 5°r¸)´}Ð EÀqÑIÔIà ÝlàAØAØAØQØBð
ð ;ð ñ ô ÷ !&¢¡¤¨!ñ ,õ tw ð # ñ # AØ !\¨!_¨a°¬d¬jUGÝ UÑ+Ô+¨L¨L¨LÔ9D1Ið $AØñ
àc1Q3hK '¨!¨!¨!¨T¨'Ô"2Ñ2ÝIa Q¡Ñ'Ô'×+Ò+¨AÑ.Ô.¨qÔ1°D´H¸ZÔ4HÒHàaDð 1Q3iØAq6l SÑ(Ø q 1¨¨qªÑ1Ô41Ø q 1¨¨qªÑ1Ô41ÝK¥¤ °Ñ!3Ô!3°Q¸¸1¸aÐ @ÑAÔAØHHYÑ'Ô'¨Ô*Ý Ô+¨CÑ0Ô0°Ô6¸¸Q¸Q¸QÀ¸W¼ ÑEÀqÔIàAJØð gg a¨mmOBS!Ø
bÑ)Ô)¨2¯7ª7©9¬9¬;IAv1Ø=×&Ò&Ñ(Ô(CØUFBð
NNAq "§)¢)¨A¨u°Q¬x¸!©|Ñ"<Ô"<¸b¿iºiÈÈ5ÐQRÌ8ÐVWÉ<Ñ>XÔ>XÐYÑ ZÔ ZÐ ZØ KK 3¨¡9¨cÐ"2°AÑ6Ô6Ñ 7Ô 7Ð 7Ø KK
Ñ #Ô #Ð #Ø KKNNNNàT7 DÐ(Ð(r N)F)r+ r, r- r r r rx r r r rF rF [ sO ØLð ð ð ð ð46Qð 6Qð 6Qðp9)ð 9)ð 9)ð 9)ð 9)r rF )r )Ú__doc__r Útorch.nnr Ú
utils.metricsr Úutils.torch_utilsr r
ÚModuler r1 rA rF r r r ú<module>rÈ s$ ððð ð
Ø Ð Ð Ð Ð Ð à "Ð "Ð "Ð "Ð "Ð "Ø )Ð )Ð )Ð )Ð )Ð )ð&ð &ð &ð &ð
ð ð ð ð BIñ ô ð ð"ð ð ð ð ñ ô ð ð<ð ð ð ð ñ ô ð ð4O)ð O)ð O)ð O)ð O)ñ O)ô O)ð O)ð O)ð O)r