config.json

{
    "model" : {
        "backend":              "Full Yolo",
        "input_size":           416,
        "anchors":              [1.57,1.77, 2.80,8.62, 3.60,3.36, 5.51,10.50, 9.22,11.30],
        "max_box_per_image":    10,        
        "labels":               ["helmet","person with helmet","person without helmet"]
    },

    "train": {
        "train_image_folder":   "train_image_folder\\",
        "train_annot_folder":   "train_annot_folder\\",     
          
        "train_times":          10,
        "pretrained_weights":   "",
        "batch_size":           2,
        "learning_rate":        1e-4,
        "nb_epochs":            50,
        "warmup_epochs":        0,

        "object_scale":         5.0 ,
        "no_object_scale":      1.0,
        "coord_scale":          1.0,
        "class_scale":          1.0,

        "saved_weights_name":   "full_yolo_helmet_and_person.h5",
        "debug":                true
    },

    "valid": {
        "valid_image_folder":   "",
        "valid_annot_folder":   "",

        "valid_times":          1
    }
}