__init__.py

from .boxes import nms, box_iou
from .roi_align import roi_align, RoIAlign
from .roi_pool import roi_pool, RoIPool
from .ps_roi_align import ps_roi_align, PSRoIAlign
from .ps_roi_pool import ps_roi_pool, PSRoIPool
from .poolers import MultiScaleRoIAlign
from .feature_pyramid_network import FeaturePyramidNetwork

from ._register_onnx_ops import _register_custom_op

_register_custom_op()


__all__ = [
    'nms', 'roi_align', 'RoIAlign', 'roi_pool', 'RoIPool',
    'ps_roi_align', 'PSRoIAlign', 'ps_roi_pool', 'PSRoIPool',
    'MultiScaleRoIAlign', 'FeaturePyramidNetwork'
]