U <■jg(у@s№ddlZddlZddlZddlZddlZeаdбddДГZd(ddДZedГdd ДГZ ed ГddДГZ ed ГddДГZedГddДГZedГddДГZ edГddДГZedГddДГZedГddДГZejаdбd d!ДГZed"Гd#d$ДГZed%Гd&d'ДГZdS))щNcCstjаdddбS)N┌torchvisionZIMPLZMeta)┌torch┌library┌Libraryйrr·C/tmp/pip-unpacked-wheel-xh8d94dr/torchvision/_meta_registrations.py┌get_meta_libsr┌defaultcsЗЗfddД}|S)Ncs,tjабr(tГаtttjjИГИГ|б|SйN)r┌ extensionZ_has_opsr┌impl┌getattrr┌ops)┌fnй┌op_name┌ overload_namerr┌wrappers zregister_meta..wrapperr)rrrrrr┌ register_metasrZ roi_alignc s^tаИаdбdkddДбtаИjИjkЗЗfddДбИаdб}Иаdб}Иа||||fбS)NщщcSsdSйNz$rois must have shape as Tensor[K, 5]rrrrr┌єz meta_roi_align..csdИjЫdИjЫЭSйNzMExpected tensor for input to have the same type as tensor for rois; but type · does not equal й┌dtyperй┌input┌roisrrrsr)r┌_check┌sizer┌ new_empty) rr ┌ spatial_scale┌ pooled_height┌pooled_width┌sampling_ratio┌aligned┌num_rois┌channelsrrr┌meta_roi_aligns ■ r+Z_roi_align_backwardcs0tаИjИjkЗЗfddДбИа||||fбS)NcsdИjЫdИjЫЭSйNzLExpected tensor for grad to have the same type as tensor for rois; but type rrrй┌gradr rrr.sz)meta_roi_align_backward..йrr!rr#)r.r r$r%r&┌ batch_sizer*┌height┌widthr'r(rr-r┌meta_roi_align_backward(s ■r3Zps_roi_alignc sФtаИаdбdkddДбtаИjИjkЗЗfddДбИаdб}tа|||dkdбИаdб}||||||f}Иа|бtj|tjdd НfS) NrrcSsdSrrrrrrr8rz#meta_ps_roi_align..csdИjЫdИjЫЭSrrrrrrr;sr·Cinput channels must be a multiple of pooling height * pooling width┌meta)r┌deviceйrr!r"rr#┌emptyZint32) rr r$r%r&r'r*r)┌out_sizerrr┌meta_ps_roi_align6s ■ ■ r:Z_ps_roi_align_backwardcs0tаИjИjkЗЗfddДбИа||| | fбS)NcsdИjЫdИjЫЭSr,rrr-rrr[sz,meta_ps_roi_align_backward..r/)r.r ┌channel_mappingr$r%r&r'r0r*r1r2rr-r┌meta_ps_roi_align_backwardKs ■r<Zroi_poolcsttаИаdбdkddДбtаИjИjkЗЗfddДбИаdб}Иаdб}||||f}Иа|бtj|dtjdНfS) NrrcSsdSrrrrrrrerzmeta_roi_pool..csdИjЫdИjЫЭSrrrrrrrhsrr5йr6rr7)rr r$r%r&r)r*r9rrr┌ meta_roi_poolcs ■ r>Z_roi_pool_backwardc s0tаИjИjkЗЗfddДбИа|||| fбS)NcsdИjЫdИjЫЭSr,rrr-rrrysz(meta_roi_pool_backward..r/) r.r Zargmaxr$r%r&r0r*r1r2rr-r┌meta_roi_pool_backwardss ■r?Zps_roi_poolcsФtаИаdбdkddДбtаИjИjkЗЗfddДбИаdб}tа|||dkdбИаdб}||||||f}Иа|бtj|dtjd НfS) NrrcSsdSrrrrrrrГrz"meta_ps_roi_pool..csdИjЫdИjЫЭSrrrrrrrЖsrr4r5r=r7)rr r$r%r&r*r)r9rrr┌meta_ps_roi_poolБs ■ ■ r@Z_ps_roi_pool_backwardc s0tаИjИjkЗЗfddДбИа|||| fбS)NcsdИjЫdИjЫЭSr,rrr-rrrЫsz+meta_ps_roi_pool_backward..r/) r.r r;r$r%r&r0r*r1r2rr-r┌meta_ps_roi_pool_backwardХs ■rAztorchvision::nmscsЮtаИабdkЗfddДбtаИаdбdkЗfddДбtаИабdkЗfddДбtаИаdбИаdбkЗЗfd dДбtjаб}|аб}Иj|tjd НS)NщcsdИабЫdЭS)Nz!boxes should be a 2d tensor, got ┌Dй┌dimrй┌detsrrrеrzmeta_nms..rщcsdИаdбЫЭS)Nz1boxes should have 4 elements in dimension 1, got rйr"rrFrrrжrcsdИабЫЭS)Nz"scores should be a 1d tensor, got rDr)┌scoresrrrзrrcsdИаdбЫdИаdбЫЭS)NzIboxes and scores should have same number of elements in dimension 0, got rz and rIrйrGrJrrrкrr) rr!rEr"Z_custom_opsZget_ctxZcreate_unbacked_symintr#┌long)rGrJZ iou_threshold┌ctxZnum_to_keeprrKr┌meta_nmsгs■ rNZ deform_conv2dcCs8|jddЕ\}}|jd}|jd}|а||||fбS)Nщ■ r)┌shaper#)r┌weight┌offset┌mask┌bias┌stride_h┌stride_w┌pad_h┌pad_wZdil_hZdil_wZ n_weight_grpsZ n_offset_grps┌use_maskZ out_heightZ out_widthZout_channelsr0rrr┌meta_deform_conv2d▒s rZZ_deform_conv2d_backwardcCsJ|а|jб}|а|jб}|а|jб}|а|jб}|а|jб}|||||fSr )r#rP)r.rrQrRrSrTrUrVrWrXZ dilation_hZ dilation_w┌groupsZ offset_groupsrYZ grad_inputZgrad_weightZgrad_offsetZ grad_maskZ grad_biasrrr┌meta_deform_conv2d_backward╔sr\)r )┌ functoolsrZtorch._custom_opsZ torch.libraryZtorchvision.extensionr┌ lru_cacherrr+r3r:r<r>r?r@rArZ register_fakerNrZr\rrrr┌s: