
    kh                     :   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Z3y)   )_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)4_register_onnx_opsr   boxesr   r   r   r   r   r	   r
   r   r   r   r   	ciou_lossr   deform_convr   r   	diou_lossr   
drop_blockr   r   r   r   feature_pyramid_networkr   
focal_lossr   	giou_lossr   miscr   r   r   r   r   r   poolersr    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   __all__     T/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/ops/__init__.py<module>r:      s[    3    - 4 , L L : * / p p ' 2 / * ' ?  (r8   