a ™¯DfÌBã @sVdZddlZddlZddlZddlZddlZddlmZ ddl mmZddl mZmZmZmZmZmZddlmZdZdZGdd„dƒZeed fd d„Zeefdd „Zd;dd„Zdd d!„Zd?d"d#„Z d@d$d%„Z!d&d'„Z"dAd+d,„Z#dd-d.d/ddd0eed f d1d2„Z$dBd3d4„Z%Gd5d6„d6ƒZ&Gd7d8„d8ƒZ'Gd9d:„d:ƒZ(dS)Cz Image augmentation functions éN)ÚLOGGERÚ check_versionÚcolorstrÚresample_segmentsÚsegment2boxÚ xywhn2xyxy)Úbbox_ioa)g ×£p= ß?gÉv¾Ÿ/Ý?g–C‹lçûÙ?)gZd;ßOÍ?gyé&1¬Ì?gÍÌÌÌÌÌÌ?c@s eZdZddd„Zd dd„ZdS) ÚAlbumentationsé€cCsd|_tdƒ}z¾ddl}t|jddd|j||dddd |jd d|jd d|j|d|j |d|j d d|jdd|jdd|j ddd g }|j||jddgdd|_t |d dd„|Dƒ¡¡WnHtyÞYn8ty}zt |›|›¡WYd}~n d}~00dS)Núalbumentations: rú1.0.3T©Úhard)gš™™™™™é?çð?)gÍÌÌÌÌÌì?gÃõ(\Âñ?ç)ÚheightÚwidthÚscaleÚratioÚpç{®Gáz„?©réK)Z quality_lowerrÚyoloÚclass_labels)ÚformatZlabel_fields)Zbbox_paramsú, css"|]}|jr|› dd¡VqdS©zalways_apply=False, ÚN©rÚreplace©Ú.0Úx©r$úI/data/NAS7/SABIOD/METHODE/ermites/yolo-dyni/yolov5/utils/augmentations.pyÚ +óz*Albumentations.__init__..)Ú transformrÚalbumentationsrÚ__version__ÚRandomResizedCropÚBlurZ MedianBlurÚToGrayÚ GaussNoiseÚCLAHEZRandomBrightnessContrastZRandomGammaZImageCompressionÚComposeZ BboxParamsrÚinfoÚjoinÚImportErrorÚ Exception)ÚselfÚpToGrayÚpGaussNoiseÚsizeÚprefixÚAÚTÚer$r$r%Ú__init__s* ÷ "zAlbumentations.__init__rcCsr|jrjt ¡|krj|j||dd…dd…f|dd…dfd}|dt dd„t|d|dƒDƒ¡}}||fS) Nér)ÚimageÚbboxesrr?cSsg|]\}}|g|¢‘qSr$r$)r"ÚcÚbr$r$r%Ú 4r'z+Albumentations.__call__..rr@)r(ÚrandomÚnpÚarrayÚzip)r5ÚimÚlabelsrÚnewr$r$r%Ú__call__1s,,zAlbumentations.__call__N)r )r)Ú__name__Ú __module__Ú__qualname__r=rKr$r$r$r%r s r FcCstj||||dS)N)Úinplace)ÚTFÚ normalize)r#ÚmeanÚstdrOr$r$r%rQ8srQcCs>tdƒD]0}|dd…|f|||||dd…|f<q|S)Né)Úrange)r#rRrSÚir$r$r%Údenormalize=s.rWçà?cCsè|s|s|rätj ddd¡|||gd}t t |tj¡¡\}}}|j}tjdd|jd} | |dd |¡} t | |ddd¡ |¡}t | |d dd¡ |¡}t t || ¡t ||¡t ||¡f¡} tj| tj |d dS)Néÿÿÿÿr>rTré©Údtypeé´éÿé)Údst)rErDÚuniformÚcv2ÚsplitÚcvtColorÚ COLOR_BGR2HSVr\ÚarangeÚastypeÚclipÚmergeÚLUTÚ COLOR_HSV2BGR)rHÚhgainÚsgainÚvgainÚrÚhueÚsatÚvalr\r#Zlut_hueZlut_satZlut_valZim_hsvr$r$r%Úaugment_hsvDs(rsTcCs t ||rtjntj¡}|rZtjddd}| |dd…dd…df¡|dd…dd…df<n.t |dd…dd…df¡|dd…dd…df<t ||r˜tjntj¡S)Ng@)ért)Z clipLimitZtileGridSizer) rbrdÚ COLOR_BGR2YUVÚ COLOR_RGB2YUVÚcreateCLAHEÚapplyÚequalizeHistÚ COLOR_YUV2BGRÚ COLOR_YUV2RGB)rHZclaheZbgrZyuvrAr$r$r%Ú hist_equalizeTs0.r|cCs&|jdd…\}}|dd…dd…f t¡}|j\}}}}||||d} | ¡dt| jdƒ…D]´} || \}}} }||| |}}tt d||¡ƒtt d||¡ƒ}}||||||g\}}}}|||…|| …f|||…||…f<t j ||| df||||ggdd}qh||fS)Nr_r>rXr)Úaxis)ÚshapergÚintr;ÚargsortÚroundr8rDrarEÚappend)rHrIÚhÚwÚboxesÚx1Úy1Úx2Úy2ÚsrVÚx1bÚy1bÚx2bÚy2bÚbhÚbwÚycÚxcÚx1aÚy1aÚx2aÚy2ar$r$r%Ú replicate_s*$&r—©r r ©érršršé c Cs¢|jdd…}t|tƒr ||f}t|d|d|d|dƒ}|sPt|dƒ}||f} tt|d|ƒƒtt|d|ƒƒf} |d| d|d| d}}|rÂt ||¡t ||¡}}n>|rd\}}|d|df} |d|d|d|df} |d}|d}|ddd…| kr6tj|| tj d}tt|dƒƒtt|dƒƒ} }tt|dƒƒtt|dƒƒ}}tj || |||tj|d }|| ||ffS) Nr_rr>r)rrrY©Ú interpolationçš™™™™™¹?)Úvalue)r~Ú isinstancerÚminrrEÚmodrbÚresizeÚINTER_LINEARÚcopyMakeBorderÚBORDER_CONSTANT)rHÚ new_shapeÚcolorÚautoZ scaleFillÚscaleupÚstrider~rorZ new_unpadÚdwÚdhÚtopÚbottomÚleftÚrightr$r$r%Ú letterboxps. " (" ""r²r$é ržr©rrc Csf|jd|dd} |jd|dd} t d¡}|jdd|d<|jdd|d<t d¡}t ||¡|d<t ||¡|d<t d¡} t ||¡}t d|d|¡}tj|d |d | dd…<t d¡}t t ||¡tj d¡|d<t t ||¡tj d¡|d <t d¡}t d|d|¡| |d<t d|d|¡| |d<||| ||}|ddks¤|ddks¤|t d¡k ¡rà|rÂtj||| | fdd}ntj||dd…| | fdd}t |ƒ}|r^t dd„|Dƒƒot |ƒ|k}t |df¡}|rÄt|ƒ}t|ƒD]Œ\}}t t |ƒdf¡}||dd…dd…f<||j}|r˜|dd…dd…f|dd…dd…fn|dd…dd…f}t|| | ƒ||<q2nJt |ddf¡}|dd…gd¢f |dd¡|dd…dd…f<||j}|r>|dd…dd…f|dd…dd…fn|dd…dd…f |d¡}|dd…gd¢f}|dd…gd¢f}t | d¡| d¡| d¡| d¡f¡ d|¡j}|dd…ddgf d| ¡|dd…ddgf<|dd…ddgf d| ¡|dd…ddgf<t|dd…dd…fj||j|r6dndd}||}|||dd…dd…f<||fS)Nrr_r>rT)rr_)r>r_)r_r)r_r>r´)ÚangleÚcenterrr])rr>)r>rrXr™)ÚdsizeÚborderValuecss|]}| ¡VqdS©N)Úanyr!r$r$r%r&Ír'z%random_perspective..é)r>r_rTr»r>r»rTr_rt)rr_r»é)r>rTéér½rrž)Úbox1Úbox2Úarea_thr)r~rEÚeyerDrarbÚgetRotationMatrix2DÚmathÚtanÚpirºÚwarpPerspectiveÚ warpAffineÚlenÚzerosrÚ enumerateÚonesr;rÚreshapeÚconcatenater¡ÚmaxrhÚbox_candidates)rHÚtargetsÚsegmentsÚdegreesÚ translaterÚshearÚperspectiveÚborderrrÚCÚPÚRÚarŠÚSr;ÚMÚnZuse_segmentsrJrVÚsegmentÚxyr#Úyr$r$r%Úrandom_perspective‘s` "" 0 B0 J4,,0râc CsXt|ƒ}|rN|rN|j\}}}t |jtj¡}tjt|ƒt||ƒdD]Ø} || || } }|| d| d|| d| df}t ||dd…dd…fƒ} | dk ¡rJt || dg|¢gfd¡}| t ||dd…dd…f|dd…dd…ffd¡¡t ||| tj¡gd d t j¡qJt |d¡}t |d¡ t¡}||||<|||fS)N)ÚkrTr_r>r»r½g333333Ó?rrY)r>r>r>)rÉr~rErÊÚuint8rDÚsamplerUrrÚallrÎr‚rbÚdrawContoursrgÚint32ÚFILLEDÚflipÚbool)rHrIrÒrrÞrƒr„rAZim_newÚjÚlrŠÚboxÚioaÚresultrVr$r$r%Ú copy_pasteñs $:$rñcCsPt ¡|krL|jdd…\}}dgddgddgddgdd gd }|D]ô}t dt||ƒ¡}t dt||ƒ¡}tdt d|¡|dƒ} tdt d|¡|dƒ} t|| |ƒ}t|| |ƒ}dd „tdƒDƒ|| |…| |…f<t|ƒrV|dkrVtj | | ||gtj d} t| t|dd…dd…f||ƒƒ}||dk}qV|S)Nr_rXr>gÐ?gÀ?r»g°?rtg ?ércSsg|]}t dd¡‘qS)é@é¿)rDÚrandint)r"Ú_r$r$r%rCr'zcutout..rTg¸…ëQ¸ž?r[r½g333333ã?) rDr~rõrrÏr¡rUrÉrErFÚfloat32rr)rHrIrrƒr„ÚscalesrŠZmask_hZmask_wÚxminÚyminÚxmaxÚymaxrîrïr$r$r%Úcutouts 2""rýcCsBtj dd¡}|||d| tj¡}t ||fd¡}||fS)Ng@@r>r)rErDÚbetargrärÎ)rHrIÚim2Úlabels2ror$r$r%Úmixup"srr_édç¼‰Ø—²Òœ)rEÚmaximum)r¿rÀZwh_thrZar_thrrÁÚepsÚw1Úh1Úw2Úh2Úarr$r$r%rÐ*s""rÐéà)g{®Gáz´?r)gè?gUUUUUUõ?gš™™™™™Ù?c Cs~tdƒ} zddl}ddlm}t|jddd|rÄ|j||||dg} | r^t | ›d¡qÞ|dkrx| |j |d g7} |dkr’| |j |d g7} |dkrÞt|ƒfd }| |jg|¢d‘RŽg7} n|j |d|j||dg} | |j||d |ƒg7} t | d dd„| Dƒ¡¡| | ¡WStyBt | ›d¡Yn8tyx}zt | ›|›¡WYd}~n d}~00dS)Nrr)Ú ToTensorV2rTr )rrrrz.auto augmentations are currently not supportedrrT)Úmax_size)rr)rRrSrcss"|]}|jr|› dd¡VqdSrrr!r$r$r%r&Sr'z*classify_albumentations..uIâš ï¸ not found, install with `pip install albumentations` (recommended))rr)Zalbumentations.pytorchrrr*r+rr1ZHorizontalFlipZVerticalFlipÚfloatÚColorJitterZSmallestMaxSizeÚ CenterCropÚ Normalizer2r0r3Úwarningr4)Úaugmentr8rrÚhflipÚvflipZjitterrRrSZauto_augr9r:rr;Zcolor_jitterr<r$r$r%Úclassify_albumentations2s0rcCs8t|tƒsJd|›dƒ‚t t|ƒtƒt tt¡g¡S)Nz ERROR: classify_transforms size z# must be integer, not (list, tuple)) r rr;r0rÚToTensorrÚ IMAGENET_MEANÚIMAGENET_STD)r8r$r$r%Úclassify_transforms\srcs&eZdZd‡fdd„ Zdd„Z‡ZS) Ú LetterBoxr˜Fr›cs8tƒ ¡t|tƒr||fn|\|_|_||_||_dSr¹)Úsuperr=r rrƒr„r©r«)r5r8r©r«©Ú __class__r$r%r=es zLetterBox.__init__csä|jdd…\}}tˆj|ˆj|ƒ}t||ƒt||ƒ}}ˆjr^‡fdd„||fDƒnˆjˆj}}t||ddƒt||ddƒ} } tjˆjˆjdfd|jd}t j |||ft jd|| | |…| | |…f<|S) Nr_c3s$|]}t |ˆj¡ˆjVqdSr¹)rÄÚceilr«r!©r5r$r%r&or'z%LetterBox.__call__..ržrTršr[rœ)r~r¡rƒr„rr©rEÚfullr\rbr£r¤)r5rHÚimhÚimwrorƒr„ÚhsÚwsr®r°Zim_outr$r r%rKks**.zLetterBox.__call__)r˜Fr›©rLrMrNr=rKÚ __classcell__r$r$rr%rcsrcs&eZdZd‡fdd„ Zdd„Z‡ZS)rr cs,tƒ ¡t|tƒr||fn|\|_|_dSr¹)rr=r rrƒr„)r5r8rr$r%r=xs zCenterCrop.__init__cCsh|jdd…\}}t||ƒ}||d||d}}tj||||…|||…f|j|jftjdS)Nr_rœ)r~r¡rbr£r„rƒr¤)r5rHr"r#Úmr®r°r$r$r%rK|s zCenterCrop.__call__)r r&r$r$rr%rvsrcs&eZdZd‡fdd„ Zdd„Z‡ZS)rFcstƒ ¡||_dSr¹)rr=Úhalf)r5r)rr$r%r=…s zToTensor.__init__cCsFt | d¡ddd…¡}t |¡}|jr2| ¡n| ¡}|d}|S)N)r_rr>rYgào@)rEÚascontiguousarrayÚ transposeÚtorchÚ from_numpyr)r)r5rHr$r$r%rK‰s zToTensor.__call__)Fr&r$r$rr%rƒsr)rXrXrX)TF)r˜r™TFTr›)r$r$r³ržržr³rr´)rX)rX)r_rržr)r))Ú__doc__rÄrDrbÚnumpyrEr,Ztorchvision.transformsÚ transformsr;Z!torchvision.transforms.functionalÚ functionalrPÚ utils.generalrrrrrrÚ utils.metricsrrrr rQrWrsr|r—r²rârñrýrrÐrrrrrr$r$r$r%Ús\ " "ø ` ö *