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