Spaces:
Sleeping
Sleeping
File size: 1,648 Bytes
749745d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
# 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",
]
|