Hugging Face's logo

ontocord
/
frcnn

Bounding box detection
PyTorch
Model card Files
xet
 Community
Ontocord.AI commited on
Commit
e8cc13d
·
1 Parent(s): 6e696e8

Create config.jsonl

Browse files
Files changed (1) hide show
  1. config.jsonl +164 -0
config.jsonl ADDED
@@ -0,0 +1,164 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "ANCHOR_GENERATOR": {
3
+ "ANGLES": [
4
+ [
5
+ -90.0,
6
+ 0.0,
7
+ 90.0
8
+ ]
9
+ ],
10
+ "ASPECT_RATIOS": [
11
+ [
12
+ 0.5,
13
+ 1.0,
14
+ 2.0
15
+ ]
16
+ ],
17
+ "OFFSET": 0,
18
+ "SIZES": [
19
+ [
20
+ 64,
21
+ 128,
22
+ 256,
23
+ 512
24
+ ]
25
+ ]
26
+ },
27
+ "BACKBONE": {
28
+ "FREEZE_AT": 2
29
+ },
30
+ "INPUT": {
31
+ "FORMAT": "BGR",
32
+ "MAX_SIZE_TEST": 1333,
33
+ "MAX_SIZE_TRAIN": 1333,
34
+ "MIN_SIZE_TEST": 800,
35
+ "MIN_SIZE_TRAIN": 640
36
+ },
37
+ "MAX_DETECTIONS": 36,
38
+ "MIN_DETECTIONS": 36,
39
+ "MODEL": {
40
+ "CHKPOINT": "",
41
+ "DEVICE": "cpu",
42
+ "LOAD_PROPOSALS": false,
43
+ "MAX_POOL": true,
44
+ "PIXEL_MEAN": [
45
+ 102.9801,
46
+ 115.9465,
47
+ 122.7717
48
+ ],
49
+ "PIXEL_STD": [
50
+ 1.0,
51
+ 1.0,
52
+ 1.0
53
+ ]
54
+ },
55
+ "PAD_VALUE": 0.0,
56
+ "PROPOSAL_GENERATOR": {
57
+ "HIDDEN_CHANNELS": 512,
58
+ "MIN_SIZE": 0.0
59
+ },
60
+ "RESNETS": {
61
+ "DEPTH": 101,
62
+ "NORM": "BN",
63
+ "NUM_GROUPS": 1,
64
+ "OUT_FEATURES": [
65
+ "res4"
66
+ ],
67
+ "RES2_OUT_CHANNELS": 256,
68
+ "RES5_DILATION": 1,
69
+ "STEM_OUT_CHANNELS": 64,
70
+ "STRIDE_IN_1X1": true,
71
+ "WIDTH_PER_GROUP": 64
72
+ },
73
+ "ROI_BOX_HEAD": {
74
+ "ATTR": true,
75
+ "BBOX_REG_WEIGHTS": [
76
+ 10.0,
77
+ 10.0,
78
+ 5.0,
79
+ 5.0
80
+ ],
81
+ "CLS_AGNOSTIC_BBOX_REG": false,
82
+ "CONV_DIM": 256,
83
+ "FC_DIM": 1024,
84
+ "NUM_ATTRS": 400,
85
+ "NUM_CONV": 0,
86
+ "NUM_FC": 0,
87
+ "POOLER_RESOLUTION": 14,
88
+ "POOLER_SAMPLING_RATIO": 0,
89
+ "RES5HALVE": false,
90
+ "SMOOTH_L1_BETA": 0.0
91
+ },
92
+ "ROI_HEADS": {
93
+ "IN_FEATURES": [
94
+ "res4"
95
+ ],
96
+ "IOU_LABELS": [
97
+ 0,
98
+ 1
99
+ ],
100
+ "IOU_THRESHOLDS": [
101
+ 0.5
102
+ ],
103
+ "NMS_THRESH_TEST": [
104
+ 0.5,
105
+ 0.6,
106
+ 0.7
107
+ ],
108
+ "NUM_CLASSES": 1600,
109
+ "POSITIVE_FRACTION": 0.25,
110
+ "PROPOSAL_APPEND_GT": true,
111
+ "SCORE_THRESH_TEST": 0.2
112
+ },
113
+ "RPN": {
114
+ "BATCH_SIZE_PER_IMAGE": 1,
115
+ "BBOX_REG_WEIGHTS": [
116
+ 1.0,
117
+ 1.0,
118
+ 1.0,
119
+ 1.0
120
+ ],
121
+ "BOUNDARY_THRESH": -1.0,
122
+ "IN_FEATURES": [
123
+ "res4"
124
+ ],
125
+ "IOU_LABELS": [
126
+ 0.0,
127
+ -1.0,
128
+ 1.0
129
+ ],
130
+ "IOU_THRESHOLDS": [
131
+ 0.3,
132
+ 0.7
133
+ ],
134
+ "LOSS_WEIGHT": 1.0,
135
+ "NMS_THRESH": 0.7,
136
+ "POSITIVE_FRACTION": 0.5,
137
+ "POST_NMS_TOPK_TEST": 300,
138
+ "POST_NMS_TOPK_TRAIN": 2000,
139
+ "PRE_NMS_TOPK_TEST": 6000,
140
+ "PRE_NMS_TOPK_TRAIN": 12000,
141
+ "SMOOTH_L1_BETA": 0.0
142
+ },
143
+ "SIZE_DIVISIBILITY": 0,
144
+ "SOLVER": {
145
+ "BASE_LR": 0.002,
146
+ "BIAS_LR_FACTOR": 1.0,
147
+ "CHECKPOINT_PERIOD": 5000,
148
+ "GAMMA": 0.1,
149
+ "IMS_PER_BATCH": 1.0,
150
+ "LR_SCHEDULER_NAME": "warmupmultisteplr",
151
+ "MAX_ITER": 270000,
152
+ "MOMENTUM": 0.9,
153
+ "STEPS": [
154
+ 210000,
155
+ 250000
156
+ ],
157
+ "WARMUP_FACTOR": 0.001,
158
+ "WARMUP_ITERS": 1000,
159
+ "WARMUP_METHOD": "linear",
160
+ "WEIGHT_DECAY": 0.0001,
161
+ "WEIGHT_DECAY_BIAS": 0.0,
162
+ "WEIGHT_DECAY_NORM": 0.0
163
+ }
164
+ }