__init__.py

from .boxes import nms, box_iou
from .roi_align import roi_align, RoIAlign
from .roi_pool import roi_pool, RoIPool
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',
    'MultiScaleRoIAlign', 'FeaturePyramidNetwork'
]