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
Showing
with 280 additions and 20 deletions
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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),
......
......@@ -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% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment