coreml-community/ControlNet-v1-1-Annotators-cpu
15
1# Copyright (c) Facebook, Inc. and its affiliates.2from .batch_norm import FrozenBatchNorm2d, get_norm, NaiveSyncBatchNorm, CycleBatchNormList3from .deform_conv import DeformConv, ModulatedDeformConv4from .mask_ops import paste_masks_in_image5from .nms import batched_nms, batched_nms_rotated, nms, nms_rotated6from .roi_align import ROIAlign, roi_align7from .roi_align_rotated import ROIAlignRotated, roi_align_rotated8from .shape_spec import ShapeSpec9from .wrappers import (10 BatchNorm2d,11 Conv2d,12 ConvTranspose2d,13 cat,14 interpolate,15 Linear,16 nonzero_tuple,17 cross_entropy,18 empty_input_loss_func_wrapper,19 shapes_to_tensor,20 move_device_like,21)22from .blocks import CNNBlockBase, DepthwiseSeparableConv2d23from .aspp import ASPP24from .losses import ciou_loss, diou_loss25 26__all__ = [k for k in globals().keys() if not k.startswith("_")]27 