yolo_v4_tflite / config.json
SamMorgan's picture
Update config.json
bb76c2b
raw
history blame
901 Bytes
{"YOLO": {"CLASSES": "./data/classes/coco.names", "ANCHORS": [12, 16, 19, 36, 40, 28, 36, 75, 76, 55, 72, 146, 142, 110, 192, 243, 459, 401], "ANCHORS_V3": [10, 13, 16, 30, 33, 23, 30, 61, 62, 45, 59, 119, 116, 90, 156, 198, 373, 326], "ANCHORS_TINY": [23, 27, 37, 58, 81, 82, 81, 82, 135, 169, 344, 319], "STRIDES": [8, 16, 32], "STRIDES_TINY": [16, 32], "XYSCALE": [1.2, 1.1, 1.05], "XYSCALE_TINY": [1.05, 1.05], "ANCHOR_PER_SCALE": 3, "IOU_LOSS_THRESH": 0.5}, "TRAIN": {"ANNOT_PATH": "./data/dataset/val2017.txt", "BATCH_SIZE": 2, "INPUT_SIZE": 416, "DATA_AUG": true, "LR_INIT": 0.001, "LR_END": 1e-06, "WARMUP_EPOCHS": 2, "FISRT_STAGE_EPOCHS": 20, "SECOND_STAGE_EPOCHS": 30}, "TEST": {"ANNOT_PATH": "./data/dataset/val2017.txt", "BATCH_SIZE": 2, "INPUT_SIZE": 416, "DATA_AUG": false, "DECTECTED_IMAGE_PATH": "./data/detection/", "SCORE_THRESHOLD": 0.25, "IOU_THRESHOLD": 0.5}{"model_type":"detr"}}