core/config.json.example at master · affinis-lab/core · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
{
    "agent": "stateful_imitation_learning_agent",
    "models" :
    {
      "road_seg_module": {
        "enabled":               true,
        "pretrained-model":     "road-seg-model.h5",
        "image-size": {
            "width": 800,
            "height": 288
        },
        "image-channels": 3
      },

      "traffic_light_module": {
        "enabled":               true,
        "pretrained-model":      "traffic-light-model.h5",
        "image-size": {
            "width": 416,
            "height": 416
        },
        "anchors":               [0.24,0.79, 0.80,2.12],
        "max_obj":               1,
        "num_classes":           2,
        "classes":               ["go", "stop"],
        "object-threshold":      0.19,
        "nms-threshold":         0.01,
        "grid_h":                13,
        "grid_w":                13
      },

      "car_module": {
        "enabled":               false,
        "image-size": {
            "width": 416,
            "height": 416
        },
        "anchors":               [0.57273, 0.677385, 1.87446, 2.06253],
        "max_obj":               7,
        "grid_h":                13,
        "grid_w":                13,
        "num_classes":           1,
        "classes":               ["car"]
      },

      "person_module": {
        "enabled":               false,
        "image-size": {
            "width": 416,
            "height": 416
        },
        "anchors":               [0.57273, 0.677385, 1.87446, 2.06253],
        "max_obj":               7,
        "grid_h":                13,
        "grid_w":                13,
        "num_classes":           1,
        "classes":               ["person"]
      }
    },

    "lstm": {
      "enabled": false,
      "num_episodes_train": 140,
      "num_episodes_val": 10,
      "images_per_episode": 200
    },


    "fc_after":                  false,
    "plot_core_model":           true,

    "train": {
        "data-folder":   "PATH-TO-DATA-DIR",
        "data-file":     "trainval.json",

        "pretrained_weights":   "",
        "batch_size":           32,
        "learning_rate":        1e-4,
        "nb_epochs":            200,

        "saved_weights_name":   "",
        "debug":                true
    },

    "preprocessing": {
      "input-dir": "PATH-TO-DATA-DIR",
      "input-file": "train.json",
      "output-dir": "PATH-TO-OUT-DIR",
      "output-file": "trainval.json"
    }
}