gun-config.json

{
    "model" : {
        "min_input_size":       235,
        "max_input_size":       235,
        "anchors":              [120,105, 160,285, 182,167, 268,246, 297,340, 348,180, 367,274, 373,338, 389,389],
        "labels":               ["gun"]
    },

    "train": {
        "train_image_folder":   "../gun_data_cross_validation/xxxxx/train_img/",
        "train_annot_folder":   "../gun_data_cross_validation/xxxxx/train_annot/",
        "cache_name":           "",

        "train_times":          8,
        "batch_size":           16,
        "learning_rate":        1e-4,
        "nb_epochs":            20,
        "warmup_epochs":        3,
        "ignore_thresh":        0.5,
        "gpus":                 "0",

        "grid_scales":          [1,1,1],
        "obj_scale":            5,
        "noobj_scale":          1,
        "xywh_scale":           1,
        "class_scale":          1,

        "tensorboard_dir":      "logs",
        "saved_weights_name":   "gun.h5",
        "debug":                true
    },

    "valid": {
        "valid_image_folder":   "../gun_data_cross_validation/xxxxx/test_img/",
        "valid_annot_folder":   "../gun_data_cross_validation/xxxxx/test_annot/",
        "cache_name":           "",

        "valid_times":          1
    }
}