-
-
Notifications
You must be signed in to change notification settings - Fork 71
/
raccoon_detector.json
37 lines (37 loc) · 1.39 KB
/
raccoon_detector.json
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
{
"model" : {
"type": "Detector",
"architecture": "MobileNet5_0",
"input_size": [240, 320],
"anchors": [[[0.76120044, 0.57155991], [0.6923348, 0.88535553], [0.47163042, 0.34163313]]],
"labels": ["raccoon"],
"obj_thresh" : 0.5,
"iou_thresh" : 0.5,
"coord_scale" : 2.0,
"object_scale" : 2.0,
"no_object_scale" : 1.0
},
"weights" : {
"full": "",
"backend": "imagenet"
},
"train" : {
"actual_epoch": 50,
"train_image_folder": "/home/ubuntu/datasets/raccoon_detector/imgs",
"train_annot_folder": "/home/ubuntu/datasets/raccoon_detector/anns",
"train_times": 2,
"valid_image_folder": "/home/ubuntu/datasets/raccoon_detector/imgs_validation",
"valid_annot_folder": "/home/ubuntu/datasets/raccoon_detector/anns_validation",
"valid_times": 2,
"valid_metric": "recall",
"batch_size": 4,
"learning_rate": 1e-4,
"saved_folder": "raccoon_detector",
"first_trainable_layer": "",
"augmentation": true,
"is_only_detect" : false
},
"converter" : {
"type": ["k210"]
}
}