-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.json
47 lines (47 loc) · 1.24 KB
/
config.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
{
"model_params": {
"architecture": "abr-net",
"input_shape": [256, 256, 3],
"output_activation": "sigmoid",
"learning_rate": 0.001,
"batch_size": 32,
"epochs": 50,
"l2_regularization": 0.0005
},
"training_params": {
"train_data_dir": "./data/train",
"validation_data_dir": "./data/validation",
"save_model_dir": "./models",
"tensorboard_log_dir": "./logs",
"checkpoint_dir": "./checkpoints",
"train_steps_per_epoch": 100,
"validation_steps_per_epoch": 50
},
"data_preprocessing": {
"image_height": 256,
"image_width": 256,
"data_augmentation": true,
"normalize_images": true,
"augmentation_params": {
"flip_left_right": true,
"flip_up_down": false,
"random_brightness": true,
"brightness_max_delta": 0.2,
"random_saturation": true,
"saturation_range": [0.7, 1.3],
"random_hue": true,
"hue_max_delta": 0.1
}
},
"evaluation_params": {
"batch_size": 32
},
"simulation_params": {
"unreal_engine_project_dir": "/path/to/unreal/project",
"simulation_config": "./configs/simulation_config.json"
},
"opencv_params": {
"masking_techniques": ["thresholding", "contours"],
"mask_save_dir": "./masks"
}
}