Chen Jiayu e61488d319
Support ViTDet algo (#35)
* adapt mmlab modules
* add vitdet
* support conv aggregation
* modify vitdet load pretrained
* support fp16
* modify defaultformatbundle
* modify aug
* bugfix sampler
* bugfix mmresize
* bugfix fp16&nonetype
* bugfix filterannotation
* support dlc
* bugfix dist
* bugfix detsourcecoco
* smodify mmdet_parse_losses
* bugfix nan
* bugfix eval
* bugfix data=nonetype
* modify resize_embed
* support vitdet_conv
* add vitdet_conv init_weight
* add test_vitdet
* uniform rand_another
* uniform use fp16 method
* add test_fp16

Co-authored-by: jiangnana.jnn <jiangnana.jnn@alibaba-inc.com>
2022-06-10 21:49:32 +08:00

17 lines
708 B
Python

# Copyright (c) Alibaba, Inc. and its affiliates.
# isort:skip_file
# yapf:disable
from .mm_transforms import (
LoadAnnotations, LoadImageFromFile, LoadMultiChannelImageFromFiles,
MMFilterAnnotations, MMMixUp, MMMosaic, MMMultiScaleFlipAug, MMNormalize,
MMPad, MMPhotoMetricDistortion, MMRandomAffine, MMRandomCrop, MMRandomFlip,
MMResize, MMToTensor, NormalizeTensor)
__all__ = [
'MMToTensor', 'NormalizeTensor', 'MMMosaic', 'MMMixUp', 'MMRandomAffine',
'MMPhotoMetricDistortion', 'MMResize', 'MMRandomFlip', 'MMPad',
'MMNormalize', 'LoadImageFromFile', 'LoadMultiChannelImageFromFiles',
'LoadAnnotations', 'MMMultiScaleFlipAug', 'MMRandomCrop',
'MMFilterAnnotations'
]