a Å þd)-ã@s<ddlZddlZddlZddlmZddlmZddlmZm Z m Z mZd%dd„Zdd„Z d&dd„ZGd d„dejƒZGdd„dejƒZGdd„dejƒZGdd„dejƒZGdd„dejƒZGdd„dejƒZGdd„dejƒZGdd„dejƒZGdd„dejƒZGdd „d ejƒZGd!d"„d"ejƒZGd#d$„d$ƒZdS)'éN)Únn©Ú letterbox)Úmake_divisibleÚnon_max_suppressionÚscale_coordsÚ xyxy2xywhcCs,|dur(t|tƒr|dndd„|Dƒ}|S)NécSsg|]}|d‘qS)r ©©Ú.0Úxr r új/var/www/html/stable-diffusion-webui/repositories/CodeFormer/facelib/detection/yolov5face/models/common.pyÚ ózautopad..)Ú isinstanceÚint)ÚkÚpr r rÚautopads rcCsV|j ¡\}}}}tj||dd}| |||||¡}t |dd¡ ¡}| |d||¡S)NÚtrunc)Ú rounding_modeér éÿÿÿÿ)ÚdataÚsizeÚtorchÚdivÚviewÚ transposeÚ contiguous)r ÚgroupsÚ batchsizeÚnum_channelsÚheightÚwidthÚchannels_per_groupr r rÚchannel_shuffles r'rTc Cst||||t ||¡|dS)N)ÚgÚact)ÚConvÚmathÚgcd)Úc1Úc2rÚsr)r r rÚDWConv%sr0cs.eZdZd ‡fdd„ Zdd„Zdd „Z‡ZS)r*rNTc sdtƒ ¡tj||||t||ƒ|dd|_t |¡|_|durFt ¡nt |tj ƒrV|nt ¡|_dS)NF)r!ÚbiasT) ÚsuperÚ__init__rÚConv2drÚconvÚBatchNorm2dÚbnÚSiLUrÚModuleÚIdentityr)©Úselfr-r.rr/rr(r)©Ú __class__r rr3,s z Conv.__init__cCs| | | |¡¡¡S©N)r)r7r5©r<r r r rÚforward2szConv.forwardcCs| | |¡¡Sr?)r)r5r@r r rÚfuseforward5szConv.fuseforward)rrNrT)Ú__name__Ú __module__Ú__qualname__r3rArBÚ __classcell__r r r=rr**sr*cs&eZdZd ‡fdd„ Zdd „Z‡ZS)Ú StemBlockér NrTcsxtƒ ¡t|||||||ƒ|_t||ddddƒ|_t|d|dddƒ|_tjdddd|_t|d|dddƒ|_ dS)Nr rrrHT)Úkernel_sizeÚstrideÚ ceil_mode) r2r3r*Ústem_1Ústem_2aÚstem_2brÚ MaxPool2dÚstem_2pÚstem_3r;r=r rr3:s zStemBlock.__init__cCs>| |¡}| |¡}| |¡}| |¡}| t ||fd¡¡S)Nr)rLrMrNrPrQrÚcat)r<r Z stem_1_outZstem_2a_outZstem_2b_outZstem_2p_outr r rrABs zStemBlock.forward)rHr NrT©rCrDrEr3rArFr r r=rrG9srGcs&eZdZd‡fdd„ Zdd„Z‡ZS) Ú BottleneckTrçà?csLtƒ ¡t||ƒ}t||ddƒ|_t||dd|d|_|oD||k|_dS)NrrH©r()r2r3rr*Úcv1Úcv2Úadd)r<r-r.Úshortcutr(ÚeÚc_r=r rr3Ls zBottleneck.__init__cCs*|jr|| | |¡¡S| | |¡¡Sr?)rYrXrWr@r r rrASszBottleneck.forward)TrrUrSr r r=rrTJsrTcs&eZdZd‡fdd„ Zdd„Z‡ZS) Ú BottleneckCSPrTrUcs¬tƒ ¡t||ƒ‰t|ˆddƒ|_tj|ˆdddd|_tjˆˆdddd|_tdˆ|ddƒ|_ t dˆ¡|_tjddd|_ tj‡‡‡fdd „t|ƒDƒŽ|_dS) NrF)r1r gš™™™™™¹?T)Úinplacec3s |]}tˆˆˆˆddVqdS©çð?)r[N©rT©rÚ_©r\r(rZr rÚ brz)BottleneckCSP.__init__..)r2r3rr*rWrr4rXÚcv3Úcv4r6r7Ú LeakyReLUr)Ú SequentialÚrangeÚm©r<r-r.ÚnrZr(r[r=rdrr3Ys zBottleneckCSP.__init__c CsD| | | |¡¡¡}| |¡}| | | tj||fdd¡¡¡S©Nr©Údim) rfrkrWrXrgr)r7rrR)r<r Úy1Úy2r r rrAds zBottleneckCSP.forward)rTrrUrSr r r=rr]Wsr]cs&eZdZd‡fdd„ Zdd„Z‡ZS) ÚC3rTrUcsntƒ ¡t||ƒ‰t|ˆddƒ|_t|ˆddƒ|_tdˆ|dƒ|_tj‡‡‡fdd„t |ƒDƒŽ|_ dS)Nrr c3s |]}tˆˆˆˆddVqdSr_rarbrdr rrerrzC3.__init__..)r2r3rr*rWrXrfrrirjrkrlr=rdrr3ls zC3.__init__cCs*| tj| | |¡¡| |¡fdd¡Srn)rfrrRrkrWrXr@r r rrAtsz C3.forward)rTrrUrSr r r=rrsjsrscs2eZdZ‡fdd„Zed dd„ƒZdd „Z‡ZS)ÚShuffleV2Blockcstƒ ¡d|krdks(ntdƒ‚||_|d}|jdkrŠt |j||d|jddt |¡tj||dddddt |¡t ¡¡|_ n t ¡|_ t tj|jdkrª|n||dddddt |¡t ¡|j||d|jddt |¡tj||dddddt |¡t ¡¡|_dS) NrrHzillegal stride valuer ©rIrJÚpaddingrF)rIrJrvr1)r2r3Ú ValueErrorrJrriÚdepthwise_convr6r4r8Úbranch1Úbranch2)r<ÚinpÚouprJÚbranch_featuresr=r rr3ys> û úñzShuffleV2Block.__init__rrFc Cstj|||||||dS)N)r1r!)rr4)ÚiÚorIrJrvr1r r rrxŸszShuffleV2Block.depthwise_convcCsb|jdkr6|jddd\}}tj|| |¡fdd}ntj| |¡| |¡fdd}t|dƒ}|S)Nrr ro)rJÚchunkrrRrzryr')r<r Úx1Úx2Úoutr r rrA£s zShuffleV2Block.forward)rrF)rCrDrEr3ÚstaticmethodrxrArFr r r=rrtxs&rtcs&eZdZd‡fdd„ Zdd„Z‡ZS)ÚSPP©éé é csXtƒ ¡|d}t||ddƒ|_t|t|ƒd|ddƒ|_t dd„|Dƒ¡|_dS)Nr rcSs g|]}tj|d|dd‘qS)rr ru)rrOrr r rr´rz SPP.__init__..) r2r3r*rWÚlenrXrÚ ModuleListrk)r<r-r.rr\r=r rr3¯s zSPP.__init__cs2| ˆ¡‰| t ˆg‡fdd„|jDƒd¡¡S)Ncsg|]}|ˆƒ‘qSr r )rrk©r r rr¸rzSPP.forward..r)rWrXrrRrkr@r rŒrrA¶s zSPP.forward)r†rSr r r=rr…sr…cs&eZdZd‡fdd„ Zdd„Z‡ZS) ÚFocusrNTcs(tƒ ¡t|d||||||ƒ|_dS)Né)r2r3r*r5r;r=r rr3½s zFocus.__init__c Csr| t |dddd…ddd…f|dddd…ddd…f|dddd…ddd…f|dddd…ddd…fgd¡¡S)N.r r)r5rrRr@r r rrAÁsz Focus.forward)rrNrTrSr r r=rr»srcs&eZdZd‡fdd„ Zdd„Z‡ZS)ÚConcatrcstƒ ¡||_dSr?)r2r3Úd)r<Ú dimensionr=r rr3Çs zConcat.__init__cCst ||j¡Sr?)rrRrr@r r rrAËszConcat.forward)rrSr r r=rrÅsrc@s eZdZdZdZdZdd„ZdS)ÚNMSçÐ?çÍÌÌÌÌÌÜ?NcCst|d|j|j|jdS)Nr©Z conf_thresZ iou_thresÚclasses)rÚconfÚiour–r@r r rrAÕszNMS.forward)rCrDrEr—r˜r–rAr r r rr’Ïsr’cs>eZdZdZdZdZdZ‡fdd„Zdd„Zdd d„Z ‡Z S) Ú AutoShapeé€r“r”Ncstƒ ¡| ¡|_dSr?)r2r3ÚevalÚmodel)r<rœr=r rr3às zAutoShape.__init__cCstdƒ|S)Nz'autoShape already enabled, skipping... )Úprint©r<r r rÚ autoshapeäszAutoShape.autoshapeFc s2tˆj ¡ƒ}t|tjƒr6ˆ | |j¡ |¡||¡St|t ƒrLt |ƒ|fnd|gf\}}gg}‰t|ƒD]°\}} t | ¡} | jddkr–| d¡} | jdkrº| dd…dd…dd…fnt | dd…dd…dfd¡} | jdd…} | | ¡|t| ƒ‰ˆ ‡fdd„| Dƒ¡| ||<ql‡fd d„t ˆd¡ d¡Dƒ‰‡fd d„|Dƒ}|dkrft |d¡n |dd}t | d¡¡}t |¡ |j¡ |¡d}t ¡"ˆ |||¡d}Wdƒn1sÐ0Yt|ˆjˆjˆjd }t|ƒD]*}tˆ||dd…dd…f||ƒqøt||ˆjƒS)Nrrr‡)rr rrHr csg|]}|ˆ‘qSr r )rÚyrVr rrÿrz%AutoShape.forward..cs g|]}t|tˆj ¡ƒƒ‘qSr )rrrJÚmaxrržr rrrcsg|]}t|ˆddd‘qS)F)Ú new_shapeÚautorr)rÚim)Úshape1r rrr)rrHrr gào@r•rŽ) ÚnextrœÚ parametersrrÚTensorÚtoÚdeviceÚtype_asÚlistrŠÚ enumerateÚnpÚarrayÚshaperÚndimÚtileÚappendr¡ÚstackÚascontiguousarrayÚ from_numpyÚno_gradrr—r˜r–rjrÚ DetectionsÚnames) r<ÚimgsrÚaugmentÚprofilerrmÚshape0r~r¤r/r r r )r(r<r¥rrAès4$ B " 2(zAutoShape.forward)ršFF)rCrDrEÚimg_sizer—r˜r–r3rŸrArFr r r=rr™Ùsr™cs.eZdZd‡fdd„ Zdd„Zdd„Z‡ZS) r¸NcsŽtƒ ¡|dj‰‡fdd„|Dƒ}||_||_||_||_dd„|Dƒ|_dd„t|j|ƒDƒ|_ dd„t|j|ƒDƒ|_ t|jƒ|_dS)Nrcs4g|],‰tjg‡fdd„dDƒ¢d‘d‘ˆd‘qS)c3s|]}ˆj|VqdSr?)r°©rr~©r¤r rrerz1Detections.__init__...)rrrrr`)rª)rÚtensor)r©rrÀrrrz'Detections.__init__..cSsg|]}t|ƒ‘qSr )rrr r rrrcSsg|]\}}||‘qSr r ©rr r(r r rrrcSsg|]\}}||‘qSr r rÃr r rrr) r2r3rªrºÚpredr¹ÚxyxyÚxywhÚzipÚxyxynÚxywhnrŠrm)r<rºrÄr¹Úgnr=rÂrr3s zDetections.__init__cCs|jSr?)rmržr r rÚ__len__"szDetections.__len__csF‡fdd„tˆjƒDƒ}|D]$}dD]}t||t||ƒdƒq$q|S)Ncs*g|]"}tˆj|gˆj|gˆjƒ‘qSr )r¸rºrÄr¹r¿ržr rr'rz%Detections.tolist..)rºrÄrÅrÈrÆrÉr)rjrmÚsetattrÚgetattr)r<r rrr ržrÚtolist%s zDetections.tolist)N)rCrDrEr3rËrÎrFr r r=rr¸s r¸)N)rrT)r+Únumpyr®rrZ+facelib.detection.yolov5face.utils.datasetsrZ*facelib.detection.yolov5face.utils.generalrrrrrr'r0r9r*rGrTr]rsrtr…rrr’r™r¸r r r rÚs( 5 :