Skip to content
Snippets Groups Projects
Commit 44ae07bb authored by zhengmiao's avatar zhengmiao
Browse files

Merge branch 'linfangjian/PhotoMetricDistortion_RandomCrop' into 'refactor_dev' 

[Refactor] Refactor RandomCrop and PhotoMetricDistortion

See merge request openmmlab-enterprise/openmmlab-ce/mmsegmentation!5
parents 619bc4a1 f6d32910
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 280 additions and 20 deletions
configs/_base_/datasets/ade20k.py
+ 14
− 1
View file @ 44ae07bb
......@@ -8,7 +8,20 @@ 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='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/ade20k_640x640.py
+ 14
− 1
View file @ 44ae07bb
......@@ -8,7 +8,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations', reduce_zero_label=True),
dict(type='Resize', img_scale=(2560, 640), ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/chase_db1.py
+ 14
− 1
View file @ 44ae07bb
......@@ -9,7 +9,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
dict(type='Resize', img_scale=img_scale, ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/cityscapes.py
+ 14
− 1
View file @ 44ae07bb
......@@ -8,7 +8,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
dict(type='Resize', img_scale=(2048, 1024), ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/cityscapes_1024x1024.py
+ 14
− 1
View file @ 44ae07bb
......@@ -6,7 +6,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
dict(type='Resize', img_scale=(2048, 1024), ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/cityscapes_768x768.py
+ 14
− 1
View file @ 44ae07bb
......@@ -6,7 +6,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
dict(type='Resize', img_scale=(2049, 1025), ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/cityscapes_769x769.py
+ 14
− 1
View file @ 44ae07bb
......@@ -6,7 +6,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
dict(type='Resize', img_scale=(2049, 1025), ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/cityscapes_832x832.py
+ 14
− 1
View file @ 44ae07bb
......@@ -6,7 +6,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
dict(type='Resize', img_scale=(2048, 1024), ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/coco-stuff10k.py
+ 14
− 1
View file @ 44ae07bb
......@@ -8,7 +8,20 @@ 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='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/coco-stuff164k.py
+ 14
− 1
View file @ 44ae07bb
......@@ -8,7 +8,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
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='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/drive.py
+ 14
− 1
View file @ 44ae07bb
......@@ -9,7 +9,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
dict(type='Resize', img_scale=img_scale, ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/hrf.py
+ 14
− 1
View file @ 44ae07bb
......@@ -9,7 +9,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
dict(type='Resize', img_scale=img_scale, ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/isaid.py
+ 14
− 1
View file @ 44ae07bb
......@@ -16,7 +16,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
dict(type='Resize', img_scale=(896, 896), ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/loveda.py
+ 14
− 1
View file @ 44ae07bb
......@@ -8,7 +8,20 @@ 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='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/pascal_context.py
+ 14
− 1
View file @ 44ae07bb
......@@ -11,7 +11,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
dict(type='Resize', img_scale=img_scale, ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/pascal_context_59.py
+ 14
− 1
View file @ 44ae07bb
......@@ -11,7 +11,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations', reduce_zero_label=True),
dict(type='Resize', img_scale=img_scale, ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/pascal_voc12.py
+ 14
− 1
View file @ 44ae07bb
......@@ -8,7 +8,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
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='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/potsdam.py
+ 14
− 1
View file @ 44ae07bb
......@@ -8,7 +8,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations', reduce_zero_label=True),
dict(type='Resize', img_scale=(512, 512), ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/stare.py
+ 14
− 1
View file @ 44ae07bb
......@@ -9,7 +9,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations'),
dict(type='Resize', img_scale=img_scale, ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
configs/_base_/datasets/vaihingen.py
+ 14
− 1
View file @ 44ae07bb
......@@ -8,7 +8,20 @@ train_pipeline = [
dict(type='LoadImageFromFile'),
dict(type='LoadAnnotations', reduce_zero_label=True),
dict(type='Resize', img_scale=(512, 512), ratio_range=(0.5, 2.0)),
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
dict(
type='TransformBroadcaster',
mapping={
'img': ['img', 'gt_semantic_seg'],
'img_shape': [..., 'img_shape']
},
auto_remap=True,
share_random_params=True,
transforms=[
dict(
type='mmseg.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),
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment