Spaces:
Sleeping
Sleeping
# 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", | |
] | |