This repository has been archived by the owner on Aug 5, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathdataset_params.py
104 lines (100 loc) · 3.55 KB
/
dataset_params.py
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#!/usr/bin/env python
# -*- coding: utf-8 -*-
__all__ = ["dataset_params"]
dataset_params = {
"coco_dataset": {
"anchors": [[10, 13], [16, 30], [33, 23], [30, 61], [62, 45], [59, 119], [116, 90], [156, 198], [373, 326],],
"anchor_masks": [[6, 7, 8], [3, 4, 5], [0, 1, 2]],
"num_classes": 80,
"img_h": 416,
"img_w": 416,
},
"udacity_dataset": {
"anchors": [
[0.014583333333333334, 0.03333333333333333],
[0.017708333333333333, 0.05],
[0.022916666666666665, 0.07333333333333333],
[0.03333333333333333, 0.043333333333333335],
[0.035416666666666666, 0.11833333333333333],
[0.051041666666666666, 0.06],
[0.08020833333333334, 0.08666666666666667],
[0.12604166666666666, 0.15666666666666668],
[0.2677083333333333, 0.3466666666666667],
],
"anchor_masks": [[6, 7, 8], [3, 4, 5], [0, 1, 2]],
"num_classes": 5,
"img_h": 416,
"img_w": 416,
},
"bird_dataset": {
"anchors": [
[0.013333333333333334, 0.013666666666666667],
[0.016923076923076923, 0.027976190476190477],
[0.022203947368421052, 0.044827615904163357],
[0.025833333333333333, 0.016710875331564987],
[0.034375, 0.028125],
[0.038752362948960305, 0.07455104993043463],
[0.05092592592592592, 0.04683129325109843],
[0.06254458977407848, 0.0764872521246459],
[0.07689655172413794, 0.14613778705636743],
[0.11500570776255709, 0.09082682291666666],
[0.162109375, 0.18448023426061494],
[0.26129166666666664, 0.3815],
],
"anchor_masks": [[8, 9, 10, 11], [4, 5, 6, 7], [0, 1, 2, 3]],
"num_classes": 1,
"img_h": 608,
"img_w": 608,
},
"voc_dataset": {
"anchors": [
[0.052, 0.08],
[0.07, 0.18018018018018017],
[0.138, 0.30930930930930933],
[0.162, 0.138],
[0.224, 0.5350840978593272],
[0.336, 0.29333333333333333],
[0.41, 0.7173333333333334],
[0.686, 0.466],
[0.848, 0.88],
],
"anchor_masks": [[6, 7, 8], [3, 4, 5], [0, 1, 2]],
"num_classes": 20,
"img_h": 416,
"img_w": 416,
},
"switch_dataset": {
"anchors": [
[0.009895833333333333, 0.006481481481481481],
[0.016145833333333335, 0.011111111111111112],
[0.021875, 0.016666666666666666],
[0.030208333333333334, 0.021296296296296296],
[0.033854166666666664, 0.032407407407407406],
[0.04635416666666667, 0.026851851851851852],
[0.05572916666666667, 0.04351851851851852],
[0.07916666666666666, 0.06574074074074074],
[0.13619791666666664, 0.11481481481481481],
],
"anchor_masks": [[6, 7, 8], [3, 4, 5], [0, 1, 2]],
"num_classes": 3,
"img_h": 608,
"img_w": 608,
},
"wheat_dataset": {
"anchors": [
[0.044921875, 0.080078125],
[0.0537109375, 0.052734375],
[0.05859375, 0.033203125],
[0.0703125, 0.06640625],
[0.0771484375, 0.091796875],
[0.0908203125, 0.0458984375],
[0.0927734375, 0.1328125],
[0.1064453125, 0.0703125],
[0.142578125, 0.1025390625],
],
"anchor_masks": [[6, 7, 8], [3, 4, 5], [0, 1, 2]],
"num_classes": 1,
"img_h": 608,
"img_w": 608,
},
}