mirror of https://github.com/YifanXu74/MQ-Det.git
35 lines
1.5 KiB
Python
35 lines
1.5 KiB
Python
# 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"]
|