mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
f0262fa68e
* Fix typo in usage example * original mosaic code in mmdet * Adjust mosaic to the semantic segmentation * Remove bbox test in test_mosaic * Add unittests * Fix resize mode for seg_fields * Fix repr error * modify Mosaic docs * modify from Mosaic to RandomMosaic * Add docstring * modify Mosaic docstring * [Docs] Add a blank line before Returns: * add blank lines Co-authored-by: MeowZheng <meowzheng@outlook.com>
20 lines
952 B
Python
20 lines
952 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .compose import Compose
|
|
from .formatting import (Collect, ImageToTensor, ToDataContainer, ToTensor,
|
|
Transpose, to_tensor)
|
|
from .loading import LoadAnnotations, LoadImageFromFile
|
|
from .test_time_aug import MultiScaleFlipAug
|
|
from .transforms import (CLAHE, AdjustGamma, Normalize, Pad,
|
|
PhotoMetricDistortion, RandomCrop, RandomCutOut,
|
|
RandomFlip, RandomMosaic, RandomRotate, Rerange,
|
|
Resize, RGB2Gray, SegRescale)
|
|
|
|
__all__ = [
|
|
'Compose', 'to_tensor', 'ToTensor', 'ImageToTensor', 'ToDataContainer',
|
|
'Transpose', 'Collect', 'LoadAnnotations', 'LoadImageFromFile',
|
|
'MultiScaleFlipAug', 'Resize', 'RandomFlip', 'Pad', 'RandomCrop',
|
|
'Normalize', 'SegRescale', 'PhotoMetricDistortion', 'RandomRotate',
|
|
'AdjustGamma', 'CLAHE', 'Rerange', 'RGB2Gray', 'RandomCutOut',
|
|
'RandomMosaic'
|
|
]
|