a
    «º”hˆ  ã                   @   s2  d dl mZ d dlmZmZmZmZmZmZm	Z	m
Z
mZmZmZ d dlmZ d dlmZmZ d dlmZ d dlmZmZmZmZ d dlmZ d dlmZ d d	lmZ d d
l m!Z!m"Z"m#Z#m$Z$m%Z%m&Z& d dl'm(Z( d dl)m)Z)m*Z* d dl+m+Z+m,Z, d dl-m-Z-m.Z. d dl/m/Z/m0Z0 d dl1m1Z1m2Z2 eƒ  g d¢Z3dS )é   )Ú_register_custom_op)Úbatched_nmsÚbox_areaÚbox_convertÚbox_iouÚclip_boxes_to_imageÚcomplete_box_iouÚdistance_box_iouÚgeneralized_box_iouÚmasks_to_boxesÚnmsÚremove_small_boxes)Úcomplete_box_iou_loss)Údeform_conv2dÚDeformConv2d)Údistance_box_iou_loss)Údrop_block2dÚdrop_block3dÚDropBlock2dÚDropBlock3d)ÚFeaturePyramidNetwork)Úsigmoid_focal_loss)Úgeneralized_box_iou_loss)ÚConv2dNormActivationÚConv3dNormActivationÚFrozenBatchNorm2dÚMLPÚPermuteÚSqueezeExcitation)ÚMultiScaleRoIAlign)Úps_roi_alignÚ
PSRoIAlign)Úps_roi_poolÚ	PSRoIPool)Ú	roi_alignÚRoIAlign)Úroi_poolÚRoIPool)Ústochastic_depthÚStochasticDepth)'r   r   r   r   r   r   r   r   r   r   r
   r	   r   r$   r%   r&   r'   r    r!   r"   r#   r   r   r   r(   r)   r   r   r   r   r   r   r   r   r   r   r   r   r   N)4Z_register_onnx_opsr   Zboxesr   r   r   r   r   r   r	   r
   r   r   r   Z	ciou_lossr   Zdeform_convr   r   Z	diou_lossr   Z
drop_blockr   r   r   r   Zfeature_pyramid_networkr   Z
focal_lossr   Z	giou_lossr   Úmiscr   r   r   r   r   r   Zpoolersr   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   Ú__all__© r,   r,   úF/var/www/auris/lib/python3.9/site-packages/torchvision/ops/__init__.pyÚ<module>   s"   4 