rodrigo2019 / keras_yolo2

MIT License
46 stars 15 forks source link

The loss is hardly to reduce and the mAP is always zero #14

Closed yololo1 closed 4 years ago

yololo1 commented 5 years ago

I use the master repo and i find the loss is hardly to reduce and the mAP is always zero I use the raccoon dataset

rodrigo2019 commented 5 years ago

how many epochs did you trained?

yololo1 commented 5 years ago

how many epochs did you trained?

153. I found that,in epoch 57,the map is 0 and the loss is 10.22 in epoch 74 ,the map is 0.2 and the loss is 9.02 finally,in epoch 153 i only get a 0.41 map and the loss is 7.56 I found the new keras-loss is hard to train,is it?

this is my config, i use squeezenet { "model" : { "backend": "SqueezeNet", "input_size_w": 224, "input_size_h": 224, "gray_mode": false, "anchors": [5.68744,8.19239, 8.66723,14.83145, 10.63554,11.19373, 13.89734,16.22634, 14.38791,12.74825], "max_box_per_image": 10,
"labels": ["raccoon"] },

"parser_annotation_type":    "xml",

"train": {
    "train_csv_file":       "",
    "train_csv_base_path":  "",
    "train_image_folder":   "/Users/keras_yolo2/raccoon_dataset-master/images",
    "train_annot_folder":   "/Users/keras_yolo2/raccoon_dataset-master/annotations",     

    "callback":             null,
    "train_times":          1,
    "pretrained_weights":   "",
    "batch_size":           4,
    "learning_rate":        1e-4,
    "nb_epochs":            150,
    "warmup_epochs":        3,

    "workers":              12,
    "max_queue_size":       40,
    "early_stop":           false,
    "tensorboard_log_dir":  "./logs/1",

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

    "saved_weights_name":   "./SuperTinyYolo_plateArea32x32.h5",
    "debug":                false
},

"valid": {
    "iou_threshold":        0.5,
    "score_threshold":      0.5,
    "valid_csv_file":       "",
    "valid_csv_base_path":  "",
    "valid_image_folder":   "",
    "valid_annot_folder":   "",

    "valid_times":          1
},

"backup":{
    "create_backup":        false,
    "redirect_model":       false,
    "backup_path":          "../backup",
    "backup_prefix":        "Tiny_yolo_VOC"
}

}

Thank you,best regards!

rodrigo2019 commented 5 years ago

Didi you comapred your training with the experiencor repository?