U
    h                  '   @   s|  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  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*d+d,d-d.d/d0d1d2d3d4d5d6d7g'Z3d8S )9   )_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StochasticDepthr   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,   J/var/www/html/venv/lib/python3.8/site-packages/torchvision/ops/__init__.py<module>   sp   4 