在mmsegmentation中训练自己的数据集
先在mmse/dataset下创建一个python文件，我的名字是my_thermal_dataset.py
在其中填写下面内容
这里要注意，在设置suffix的时候，如果你的label文件和train图片的后缀不一样，记得加上，我这里的label文件有后缀_label，别忘记了
import mmcv
from mmcv.utils import print_log
import os.path as osp
from ..utils import get_root_logger
from .builder import DATASETS
from .custom import CustomDataset
@DATASETS.register_module()
class mydata(CustomDataset):
CLASSES = ('background', 'person')
PALETTE = [[0,0,0], [128,0,0]]
def __init__(self, **kwargs):
super(mydata, self).__init__(
img_suffix='.JPEG',
seg_map_suffix='_label.png',
reduce_zero_label=False,
**kwargs)
assert osp.exists(self.img_dir)
然后需要在_ _init__.py文件中添加你自己的数据集的名称
Copyright (c) OpenMMLab. All rights reserved.
from .ade import ADE20KDataset
from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset
from .chase_db1 import ChaseDB1Dataset
from .cityscapes import CityscapesDataset
from .coco_stuff import COCOStuffDataset
from .custom import CustomDataset
from .dark_zurich import DarkZurichDataset
from .dataset_wrappers import (ConcatDataset, MultiImageMixDataset,
RepeatDataset)
from .drive import DRIVEDataset
from .hrf import HRFDataset
from .isaid import iSAIDDataset
from .isprs import ISPRSDataset
from .loveda import LoveDADataset
from .night_driving import NightDrivingDataset
from .pascal_context import PascalContextDataset, PascalContextDataset59
from .potsdam import PotsdamDataset
from .stare import STAREDataset
from .voc import PascalVOCDataset
from .my_thermal_dataset import my_thermal_dataset
__all__ = [
'CustomDataset', 'build_dataloader', 'ConcatDataset', 'RepeatDataset',
'DATASETS', 'build_dataset', 'PIPELINES', 'CityscapesDataset',
'PascalVOCDataset', 'ADE20KDataset', 'PascalContextDataset',
'PascalContextDataset59', 'ChaseDB1Dataset', 'DRIVEDataset', 'HRFDataset',
'STAREDataset', 'DarkZurichDataset', 'NightDrivingDataset',
'COCOStuffDataset', 'LoveDADataset', 'MultiImageMixDataset',
'iSAIDDataset', 'ISPRSDataset', 'PotsdamDataset','my_thermal_dataset'
]
然后需要更改/mmsegmentation-master/mmseg/core/evaluation下的class_names.py文件中的class
添加两个部分
def my_thermal_dataset_classes():
return ['background', 'person']
def my_thermal_dataset_palette():
return [[0,0,0], [128,0,0]]
在数据集部分添加告一段落，后需要在config/_base_文件夹下添加相对应的文件
添加my_thermal_datasets.py
里面填写的内容我是在ade20k的配置文件的基础上修改的
需要修改对应的路径
dataset settings
dataset_type = 'my_thermal_dataset'
data_root = './thermal_data'
img_norm_cfg = dict(
mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225], to_rgb=True)
crop_size = (512, 512)
train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations', reduce_zero_label=True),
dict(type='Resize', img_scale=(2048, 512), ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(type='RandomFlip', prob=0.5),
dict(type='PhotoMetricDistortion'),
dict(type='Normalize', **img_norm_cfg),
dict(type='Pad', size=crop_size, pad_val=0, seg_pad_val=255),
dict(type='DefaultFormatBundle'),
dict(type='Collect', keys=['img', 'gt_semantic_seg']),
]
test_pipeline = [
dict(type='LoadImageFromFile'),
dict(
type='MultiScaleFlipAug',
img_scale=(2048, 512),
# img_ratios=[0.5, 0.75, 1.0, 1.25, 1.5, 1.75],
flip=False,
transforms=[
dict(type='Resize', keep_ratio=True),
dict(type='RandomFlip'),
dict(type='Normalize', **img_norm_cfg),
dict(type='ImageToTensor', keys=['img']),
dict(type='Collect', keys=['img']),
])
]
data = dict(
samples_per_gpu=4,
workers_per_gpu=4,
train=dict(
type=dataset_type,
data_root=data_root,
img_dir='img_dir/train',
ann_dir='ann_dir/train',
pipeline=train_pipeline),
val=dict(
type=dataset_type,
data_root=data_root,
img_dir='img_dir/val',
ann_dir='ann_dir/val',
pipeline=test_pipeline),
test=dict(
type=dataset_type,
data_root=data_root,
img_dir='img_dir/val',
ann_dir='img_dir/val',
pipeline=test_pipeline))
开始训练模型
GPUS=8 sh tools/slurm_train.sh dev mae_thermal configs/mae/upernet_mae-base_fp16_512×512_160k_thermal.py --work-dir work_dirs/thermal_res/
Original: https://www.cnblogs.com/yuzhoutaiyang/p/16225543.html
Author: 宇宙•太阳
Title: mmsegmentation中构造自己的数据集和数据加载部分，跑现有demo
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/806815/
转载文章受原作者版权保护。转载请注明原作者出处！