Skip to content

Commit bb3f4c6

Browse files
committed
Add dataset config for sizeRight credit card segmentation
1 parent adf0aa0 commit bb3f4c6

File tree

1 file changed

+78
-0
lines changed

1 file changed

+78
-0
lines changed
Lines changed: 78 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,78 @@
1+
# dataset settings
2+
dataset_type = 'MyCreditCardDataset'
3+
data_root = 'C:/Users/zliu/Projects/FeetReferenceModel_Mask_RCNN/aetrex_module/data/dataset'
4+
img_norm_cfg = dict(
5+
mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
6+
crop_size = (512, 512)
7+
train_pipeline = [
8+
dict(type='LoadImageFromFile'),
9+
dict(type='LoadAnnotations', reduce_zero_label=False),
10+
dict(type='Resize', img_scale=(1024, 512), ratio_range=(0.5, 2.0)),
11+
dict(type='RandomCrop', crop_size=crop_size, cat_max_ratio=0.75),
12+
dict(type='RandomFlip', prob=0.5),
13+
dict(type='PhotoMetricDistortion'),
14+
dict(type='Normalize', **img_norm_cfg),
15+
dict(type='Pad', size=crop_size, pad_val=0, seg_pad_val=255),
16+
dict(type='DefaultFormatBundle'),
17+
dict(type='Collect', keys=['img', 'gt_semantic_seg']),
18+
]
19+
test_pipeline = [
20+
dict(type='LoadImageFromFile'),
21+
dict(
22+
type='MultiScaleFlipAug',
23+
img_scale=(1024, 512),
24+
# img_ratios=[0.5, 0.75, 1.0, 1.25, 1.5, 1.75],
25+
flip=True,
26+
transforms=[
27+
dict(type='Resize', keep_ratio=True),
28+
dict(type='RandomFlip'),
29+
dict(type='Normalize', **img_norm_cfg),
30+
dict(type='ImageToTensor', keys=['img']),
31+
dict(type='Collect', keys=['img']),
32+
])
33+
]
34+
data = dict(
35+
samples_per_gpu=2,
36+
workers_per_gpu=2,
37+
train=dict(
38+
type=dataset_type,
39+
data_root=data_root,
40+
img_dir=[
41+
'russian_latest/training_card_Nov11/train/images',
42+
'russian_latest/additional_training_augment_credit_card_Dec17_430/train/images',
43+
'russian_latest/additional_training_credit_card_Dec17_430/train/images',
44+
'credit_card_data1/train/images',
45+
#'russian_latest/training_augment_card_Nov11/train/images',
46+
],
47+
ann_dir=[
48+
'russian_latest/training_card_Nov11/train/annotations',
49+
'russian_latest/additional_training_augment_credit_card_Dec17_430/train/annotations',
50+
'russian_latest/additional_training_credit_card_Dec17_430/train/annotations',
51+
'credit_card_data1/train/annotations',
52+
#'russian_latest/training_augment_card_Nov11/train/annotations',
53+
],
54+
pipeline=train_pipeline),
55+
val=dict(
56+
type=dataset_type,
57+
data_root=data_root,
58+
img_dir=[
59+
'russian_latest/training_card_Nov11/val/images',
60+
'russian_latest/additional_training_augment_credit_card_Dec17_430/val/images',
61+
'russian_latest/additional_training_credit_card_Dec17_430/val/images',
62+
'credit_card_data1/val/images',
63+
#'russian_latest/training_augment_card_Nov11/val/images',
64+
],
65+
ann_dir=[
66+
'russian_latest/training_card_Nov11/val/annotations',
67+
'russian_latest/additional_training_augment_credit_card_Dec17_430/val/annotations',
68+
'russian_latest/additional_training_credit_card_Dec17_430/val/annotations',
69+
'credit_card_data1/val/annotations',
70+
#'russian_latest/training_augment_card_Nov11/val/annotations',
71+
],
72+
pipeline=test_pipeline),
73+
test=dict(
74+
type=dataset_type,
75+
data_root=data_root,
76+
img_dir='my_test_images/images',
77+
ann_dir='my_test_images/annotations',
78+
pipeline=test_pipeline))

0 commit comments

Comments
 (0)