Arulkumar03/Wheat_HEAD_Detection_Counting_ComputerVision_Model

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