-
Notifications
You must be signed in to change notification settings - Fork 10
/
config.json
47 lines (47 loc) · 1.13 KB
/
config.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
38
39
40
41
42
43
44
45
46
47
{
"DETECTOR": "detectors.faster_rcnn",
"DETECTOR_ALG": "binary_fasterrcnn_resnet50",
"DETECTOR_WEIGHTS": "./models/detector.pth",
"USE_CLASSIFIER": true,
"CLASSIFIER": "knn",
"CLASSIFIER_WEIGHTS": "./models/classifier_resnet18.pth",
"DISTANCE": "cosine",
"MAX_DISTANCE": 0.01,
"PATH_EXEMPLARS_EMBEDDINGS": "./data/exemplars_siamese.pkl",
"LOAD_EMBEDDINGS": false,
"EMBEDDING_SIZE": 345,
"EMBEDDER_ALG": "siamese_resnet18",
"EMBEDDER_WEIGHTS": "./models/embedder.pth",
"EMBEDDER_IMG_SIZE": 100,
"EXEMPLARS_FORMAT": "jpg",
"IMAGE_RESIZE": [
100,
100
],
"MIN_CONFIDENCE": 0.9,
"AUGMENTER_PARAMS": {
"Multiply": [
0.5,
1.5
],
"GaussianBlur": [
0.4
],
"AdditiveGaussianNoise": [
51.0
],
"AffineShear": [
-25,
25
],
"AffineRotate": [
-25,
25
]
},
"DEVICE": "cpu",
"EMBEDDER_DEVICE": "cpu",
"DETECTOR_DEVICE": "cpu",
"CLASSIFIER_DEVICE": "cpu",
"CLASSIFIER_ALG": "knn"
}