o á,ËhSã@s<ddlZddlmZGdd„dejƒZGdd„dejƒZdS)éNcs,eZdZ‡fdd„Zdejfdd„Z‡ZS)Ú MaskL1Losscstt|ƒ ¡dS©N)ÚsuperrÚ__init__)Úself©Ú __class__©úY/var/www/html/scripts/venv/lib/python3.10/site-packages/easyocr/DBNet/decoders/l1_loss.pyrszMaskL1Loss.__init__ÚpredcCsV| ¡}| ¡dkr|t|dfSt |dd…df|¡| ¡|}|t|dfS)Nr)Úl1_loss)ÚsumÚitemÚdictÚtorchÚabs)rr ÚgtÚmaskÚmask_sumÚlossr r r Úforward s  &zMaskL1Loss.forward©Ú__name__Ú __module__Ú __qualname__rrÚTensorrÚ __classcell__r r rr rs rcs.eZdZd‡fdd„ Zdejfdd„Z‡ZS)Ú BalanceL1Lossç@cstt|ƒ ¡||_dSr)rrrÚnegative_ratio)rrrr r rs zBalanceL1Loss.__init__r c Csœt |dd…df|¡}||}|d|}t| ¡ƒ}ttd| ¡ƒt||jƒƒ}t | d¡|¡\} } |  ¡|} | ¡|} | | t| | dfS)zo Args: pred: (N, 1, H, W). gt: (N, H, W). mask: (N, H, W). Nrééÿÿÿÿ)r Ú nge_l1_loss) rrÚintr ÚminrÚtopkÚviewr) rr rrrÚpositiveÚnegativeÚpositive_countÚnegative_countÚ negative_lossÚ_Ú positive_lossr r r rs   þ   ÿzBalanceL1Loss.forward)rrr r rr rsr)rÚtorch.nnÚnnÚModulerrr r r r Ús