maskrcnn_benchmark/layers/__init__.py

# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
import torch

from .batch_norm import FrozenBatchNorm2d, NaiveSyncBatchNorm2d
from .misc import Conv2d, _NewEmptyTensorOp
from .misc import ConvTranspose2d
from .misc import DFConv2d
from .misc import interpolate
from .misc import Scale
from .nms import nms
from .nms import ml_nms
from .nms import soft_nms
from .roi_align import ROIAlign
from .roi_align import roi_align
from .roi_align import ROIAlignV2
from .roi_pool import ROIPool
from .roi_pool import roi_pool
from .smooth_l1_loss import smooth_l1_loss
from .sigmoid_focal_loss import SigmoidFocalLoss, TokenSigmoidFocalLoss
from .iou_loss import IOULoss, IOUWHLoss
from .deform_conv import DeformConv, ModulatedDeformConv
from .dropblock import DropBlock2D, DropBlock3D
from .evonorm import EvoNorm2d
from .dyrelu import DYReLU, swish
from .se import SELayer, SEBlock
from .dyhead import DyHead
from .set_loss import HungarianMatcher, SetCriterion

__all__ = [
    "nms",
    "ml_nms",
    "soft_nms",
    "roi_align",
    "ROIAlign",
    "roi_pool",
    "ROIPool",
    "smooth_l1_loss",
    "Conv2d",
    "ConvTranspose2d",
    "interpolate",
    "swish",
    "FrozenBatchNorm2d",
    "NaiveSyncBatchNorm2d",
    "SigmoidFocalLoss",
    "TokenSigmoidFocalLoss",
    "IOULoss",
    "IOUWHLoss",
    "Scale",
    "DeformConv",
    "ModulatedDeformConv",
    "DyHead",
    "DropBlock2D",
    "DropBlock3D",
    "EvoNorm2d",
    "DYReLU",
    "SELayer",
    "SEBlock",
    "HungarianMatcher",
    "SetCriterion",
    "ROIAlignV2",
    "_NewEmptyTensorOp",
]