NgfL dZddlmZddlmZdZeddddZedid ed d d d d d ddedd dedddddd ddedd dedddddd ddedd d ed d!d"d#d$dddedd d%ed%d&d'd(d)dedd*d+ed+d,d-d.d)dedd*d/ed/d0dd1dd gd2dd3d3dd4edd56d7ed7d8d'd9d)dgd2dd3d3dd4edd:6d;ed;d? d@edidAd@dBdCdDddEgd2dFddGddHd dIddJdKdLdMdNd3dOd3dPddQddRd4dSeddTdUdVedVdWdgd2ddd ddKddd4eddXdYedYdWdgd2dd d ddKdZddd4edd[\d]ed]d^dgd2ddd ddKddd4edd_d`daedidAdadBd^dDddEgd2dFdbdGddHd dIddcdddJdKdLdedNd3dOd3dPddQddRd4dSeddfd_dTdgdhedhdid gd2d d d ddd4eddj dkedkdldgd2ddd ddd4eddj dmedmdndogd2dpd d dddd4edqdr dsedsdtd gd2dud d dddd4eddr dvedvdwd gd2d d d ddxdd4eddr dyedyd d d d d ddddd4eddz{ d|ed|dddd d dddd3d3dd4edd}~dedddddd ddddd4edd{ dedd d gd2dd d ddd4edddj deddd gd2dd d dd=dd4dedddedddgd2dbdd dd=dd4dedddedddod d d ddd4edd deddd d d d edd*dedddddd edd*dedddddd edd*deddd"d#d$dedd*deddd'd(d)dedd*deddd-d.d)dedd*deddd-ddddedd deddddddddedd dedddddddddedd deddd d d d dddedd dedddddd dddedd dedddddd dddedd deddd"d#d$ddddedd deddd'd(d)ddddedd deddd-d.d)ddddedd dedddddd d d eddœedTdideddddodddd eddœedTdidedddddddd eddœedTdidedddd d#d$dedd͜edTdideddddddd$deddœedTdideddddd(d)dedd͜edTdiZddՄZ dS)zEfficientDet Configurations Adapted from official impl at https://github.com/google/automl/tree/master/efficientdet TODO use a different config system (OmegaConfig -> Hydra?), separate model from train specific hparams ) OmegaConf)deepcopyctj}d|_d|_d|_d|_d|_d|_d|_d|_ |j |jz dz|_ d|_ gd |_ d |_ d |_d |_d|_t#d d|_d|_d|_d|_d|_d|_d|_d|_d|_d|_d|_d|_d|_d|_d|_ d|_!d|_"d|_#d|_$d|_%d|_&d|_'d|_(d|_)d|_*|S)z$Returns a default detection configs.tf_efficientdet_d1tf_efficientnet_b1Nr Z))?r)ffffff?ffffff?)rr@sameswishgMbP?g{Gz?)epsmomentumXTFmaxnearestg?g?皙?gI@id)+rcreatename backbone_name backbone_argsbackbone_indices image_size num_classes min_level max_level num_levels num_scales aspect_ratios anchor_scalepad_typeact_type norm_layerdict norm_kwargsbox_class_repeatsfpn_cell_repeats fpn_channelsseparable_convapply_resample_bnconv_bn_relu_patterndownsample_type upsample_typeredundant_biashead_bn_level_first head_act_typefpn_name fpn_configfpn_drop_path_ratealphagammalabel_smoothing legacy_focaljit_lossdeltabox_loss_weightsoft_nmsmax_detection_pointsmax_det_per_image)hs V/var/www/html/ai-engine/env/lib/python3.11/site-packages/effdet/config/model_config.pydefault_detection_model_configsrH snA"AF*AOAOAALAMAKAK;,q0ALAL:::AOANAJAJALTC000AMAAANAA"AAAOA!AAOAJALAAGAGAANAJAGAAJ!AA H bifpn_sum)?rKrKrelu6)r9meanstdr*efficientdet_d0efficientnet_b0)rQ@r Fr)drop_path_ratezehttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/efficientdet_d0-f3276ba8.pth) rrr!r0r/r.r)r6rurlefficientdet_d1efficientnet_b1rrg?zehttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/efficientdet_d1-bb7e98fe.pthefficientdet_d2efficientnet_b2)r[pefficientdet_d3efficientnet_b3)r`efficientdet_d4efficientnet_b4)rer )rrr!r0r/r.rrUefficientdet_d5efficientnet_b5)rii efficientdetv2_dtefficientnetv2_rw_t)rg@rKbilinearTznhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/efficientdetv2_rw_dt_agc-ad8b8c96.pth)rrr!r0r/r.r'r)r4r5r6r7rrUefficientdetv2_dsefficientnetv2_rw_sznhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/efficientnetv2_rw_ds_agc-cf589293.pthresdet50resnet50reluzbhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/resdet50_416-08676892.pth) rrr!r0r/r.r)r*r6r1rrU cspresdet50rr cspresnet50r!r'r0r/r.r)r* leaky_relur8silur4r5r6r1r7rrUzbhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/cspresdet50b-386da277.pth cspresdext50 cspresnext50)rrr!r'r0r/r.r)r*r6r1r7rrUcspresdext50panpan_fazehttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/cspresdext50pan-92fdd094.pth)rrr!r'r0r/r.r)r*r9r6r1r7rrU cspdarkdet53 cspdarknet53)r rXr])rrr!r'r0r/r.r)r*r6r1r7rr rU cspdarkdet53m`r9qufpn_famishr zchttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/cspdarkdet53m-79062b2d.pthmixdet_mmixnet_m) rrr!r'r0r/r.r)r6r7rrUmixdet_lmixnet_lmobiledetv2_110dmobilenetv2_110d)r0g?) rrr!r'r0r/r.r)r*r6r7rrUmobiledetv2_120dmobilenetv2_120d8mobiledetv3_largemobilenetv3_large_100 hard_swishefficientdet_q0zehttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/efficientdet_q0-bdf1bdb5.pth) rrr!r0r/r.r)r9r6r7rrUefficientdet_q1zfhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/efficientdet_q1b-d0612140.pth)rrr!r0r/r.r)r9r4r5r6r7rrUefficientdet_q2zehttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/efficientdet_q2-0f7564e5.pthefficientdet_w0P depthwise)rTfeature_locationefficientdet_esefficientnet_esH)rrr!r'r0r/r.r)r*r6r7r1rrUefficientdet_emefficientnet_emefficientdet_lite0efficientnet_lite0) rrr!r0r/r.r*r6r7rrUtf_efficientdet_d0tf_efficientnet_b0zkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d0_34-f153e0cf.pthrrzkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d1_40-a30f94af.pthtf_efficientdet_d2tf_efficientnet_b2zkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d2_43-8107aa99.pthtf_efficientdet_d3tf_efficientnet_b3zkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d3_47-0b525f35.pthtf_efficientdet_d4tf_efficientnet_b4zkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d4_49-f56376d9.pthtf_efficientdet_d5tf_efficientnet_b5zkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d5_51-c79f9be6.pthtf_efficientdet_d6tf_efficientnet_b6rzkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d6_52-4eda3773.pth) rrr!r0r/r.r9rrUtf_efficientdet_d7)rg@zkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d7_53-6d1d7a95.pth) rrr!r0r/r.r(r9rrUtf_efficientdet_d7xtf_efficientnet_b7rzihttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d7x-f390b87c.pth) rrr!r0r/r.r(r$r9rrUtf_efficientdet_d0_apzkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d0_ap-d0cdbd0a.pth) rrr!r0r/r.rMrN fill_colorrrUtf_efficientdet_d1_apzkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d1_ap-7721d075.pthtf_efficientdet_d2_apzkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d2_ap-a2995c19.pthtf_efficientdet_d3_apzkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d3_ap-e4a2feab.pthtf_efficientdet_d4_apzkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d4_ap-f601a5fc.pthtf_efficientdet_d5_apzkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_d5_ap-3673ae5d.pthtf_efficientdet_lite0tf_efficientnet_lite0)@rg@r)rrr!r(r0r/r.rzkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_lite0-dfacfc78.pthtf_efficientdet_lite1tf_efficientnet_lite1zkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_lite1-6dc7ab30.pthtf_efficientdet_lite2tf_efficientnet_lite2)rzkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_lite2-86c5d55b.pthtf_efficientdet_lite3tf_efficientnet_lite3)rrr!r0r/r.rzkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_lite3-506852cb.pthtf_efficientdet_lite3xzlhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_lite3x-8404c57b.pthtf_efficientdet_lite4tf_efficientnet_lite4zkhttps://github.com/rwightman/efficientdet-pytorch/releases/download/v0.1/tf_efficientdet_lite4-391ddabc.pthct}|t||j|jz dz|_t |}|S)zrs A A A H#d    !%A !A !A !D 'd#... s    A !D 'd#... s    A !4D 'd#...     5A !LD 'd#...     MA !dD 'd#...     eA !xD 'd#...     yA !Nd +%oo"  d#... |OA !pd +%oo"  d#... |qA !XT  d#... p    YA !t ]#m:&ooo  R   !f# !juu!D d#....!" q p#uA !Z $%oo d#... [A !zD $%oo d#... s{A !\ $%oo d#..." ]A !~$ _$n:&ooo  R   !f# !juu !D!"d#....#$#%& r q'A !hT  %oo d#...     iA !DT  %oo d#...     EA !`T (%oo d$/// aA !~T (%oo d#... A !\d -%oo d#... ]A !zD ' d#... s    {A !V D '"  d#... tW A !v D ' d#... s    w A !R D '%oo d(*** S A !r D '%oo d#... s A !R D '%oo d#... S A !r t !* d#...     s A !P t !*d#... y    Q A !d t !*d#... y    e A !x t !*d#... y    y A !L t !*d#... y    M A !` t !*d#... y    a A !t t !*d#... y    u A !Ht !*d#... y    IA !^t !*d#... y    _A !v "*d#... w    wA !V$ $*  d#... y    WA !p$ $*  d#... y    qA !J$ $*  d#... y    KA !d$ $*  d#... y    eA !~$ $*  d#... y    A !X$ $*  d#... y    YA !v$  $-d"---   &   z    wA !N$  $-d#...   &   z    OA !f$  $-d#...   &   z    gA !~$  $-d#...   &   z    A !T 4  %-d#...   &   {    UA !l$  $-d#...   &   z    mA !H      rI