mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
77591b9e7b
## Motivation For support with reading multiple remote sensing image formats, please refer to https://gdal.org/drivers/raster/index.html. Byte, UInt16, Int16, UInt32, Int32, Float32, Float64, CInt16, CInt32, CFloat32 and CFloat64 are supported for reading and writing. Support input of two images for change detection tasks, and support the LEVIR-CD dataset. ## Modification Add LoadSingleRSImageFromFile in 'mmseg/datasets/transforms/loading.py'. Load a single remote sensing image for object segmentation tasks. Add LoadMultipleRSImageFromFile in 'mmseg/datasets/transforms/loading.py'. Load two remote sensing images for change detection tasks. Add ConcatCDInput in 'mmseg/datasets/transforms/transforms.py'. Combine images that have been separately augmented for data enhancement. Add BaseCDDataset in 'mmseg/datasets/basesegdataset.py' Base class for datasets used in change detection tasks. --------- Co-authored-by: xiexinch <xiexinch@outlook.com>
29 lines
1.5 KiB
Python
29 lines
1.5 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .formatting import PackSegInputs
|
|
from .loading import (LoadAnnotations, LoadBiomedicalAnnotation,
|
|
LoadBiomedicalData, LoadBiomedicalImageFromFile,
|
|
LoadImageFromNDArray, LoadMultipleRSImageFromFile,
|
|
LoadSingleRSImageFromFile)
|
|
# yapf: disable
|
|
from .transforms import (CLAHE, AdjustGamma, Albu, BioMedical3DPad,
|
|
BioMedical3DRandomCrop, BioMedical3DRandomFlip,
|
|
BioMedicalGaussianBlur, BioMedicalGaussianNoise,
|
|
BioMedicalRandomGamma, ConcatCDInput, GenerateEdge,
|
|
PhotoMetricDistortion, RandomCrop, RandomCutOut,
|
|
RandomMosaic, RandomRotate, RandomRotFlip, Rerange,
|
|
ResizeShortestEdge, ResizeToMultiple, RGB2Gray,
|
|
SegRescale)
|
|
|
|
# yapf: enable
|
|
__all__ = [
|
|
'LoadAnnotations', 'RandomCrop', 'BioMedical3DRandomCrop', 'SegRescale',
|
|
'PhotoMetricDistortion', 'RandomRotate', 'AdjustGamma', 'CLAHE', 'Rerange',
|
|
'RGB2Gray', 'RandomCutOut', 'RandomMosaic', 'PackSegInputs',
|
|
'ResizeToMultiple', 'LoadImageFromNDArray', 'LoadBiomedicalImageFromFile',
|
|
'LoadBiomedicalAnnotation', 'LoadBiomedicalData', 'GenerateEdge',
|
|
'ResizeShortestEdge', 'BioMedicalGaussianNoise', 'BioMedicalGaussianBlur',
|
|
'BioMedical3DRandomFlip', 'BioMedicalRandomGamma', 'BioMedical3DPad',
|
|
'RandomRotFlip', 'Albu', 'LoadSingleRSImageFromFile', 'ConcatCDInput',
|
|
'LoadMultipleRSImageFromFile'
|
|
]
|