mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
83d312e87a
* Add support for Pascal Context 59 classes (#459) * Create PascalContextDataset59 class in mmseg/datasets/pascal_context.py; * Set reduce_zero_label=True for train_pipeline and PascalContextDataset59; * Add some configs for Pascal-Context 59 classes training and testing; * Try to solve the problem about "fence(IoU)=nan grass(IoU)=0"; * Continue(1): Try to solve the problem about "fence(IoU)=nan grass(IoU)=0"; * ignore files and folders named tempxxx; * Continue(2): Try to solve the problem about "fence(IoU)=nan grass(IoU)=0"; * Modify the calculation of IoU; * Modify the CLASSES order of PascalContextDataset; * Add "fcn", "deeplabv3", "deeplabv3+", "pspnet" config file for model training based on PascalContextDataset59; Add some ignore items in ".gitignore"; * fix the bug "test_cfg specified in both outer field and model field " of pspnet config file; * * Clean unnecessary codes; * Add weighs link, config link, log link and evaluation results about PascalContextDataset59 in README.md * Add command line argument: "-p | --port", this arg can change the transmit port when you transmit data to distributed machine. * * Remove rebundant config files; * Remove "-p|--port" command argument; Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
20 lines
798 B
Python
20 lines
798 B
Python
from .ade import ADE20KDataset
|
|
from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset
|
|
from .chase_db1 import ChaseDB1Dataset
|
|
from .cityscapes import CityscapesDataset
|
|
from .custom import CustomDataset
|
|
from .dataset_wrappers import ConcatDataset, RepeatDataset
|
|
from .drive import DRIVEDataset
|
|
from .hrf import HRFDataset
|
|
from .pascal_context import PascalContextDataset, PascalContextDataset59
|
|
from .stare import STAREDataset
|
|
from .voc import PascalVOCDataset
|
|
|
|
__all__ = [
|
|
'CustomDataset', 'build_dataloader', 'ConcatDataset', 'RepeatDataset',
|
|
'DATASETS', 'build_dataset', 'PIPELINES', 'CityscapesDataset',
|
|
'PascalVOCDataset', 'ADE20KDataset', 'PascalContextDataset',
|
|
'PascalContextDataset59', 'ChaseDB1Dataset', 'DRIVEDataset', 'HRFDataset',
|
|
'STAREDataset'
|
|
]
|