Upload 2 files
Browse files
configs/rtdetr/rtdetr_r101vd_6x_cococope12.yml
ADDED
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
__include__: [
|
2 |
+
'../dataset/coco_detection.yml',
|
3 |
+
'../runtime.yml',
|
4 |
+
'./include/dataloader.yml',
|
5 |
+
'./include/optimizer.yml',
|
6 |
+
'./include/rtdetr_r50vdcope12.yml',
|
7 |
+
]
|
8 |
+
|
9 |
+
PResNet:
|
10 |
+
depth: 101
|
11 |
+
|
12 |
+
|
13 |
+
HybridEncoder:
|
14 |
+
# intra
|
15 |
+
hidden_dim: 384
|
16 |
+
dim_feedforward: 2048
|
17 |
+
|
18 |
+
|
19 |
+
RTDETRTransformer:
|
20 |
+
feat_channels: [384, 384, 384]
|
21 |
+
|
22 |
+
|
23 |
+
optimizer:
|
24 |
+
type: AdamW
|
25 |
+
params:
|
26 |
+
-
|
27 |
+
params: 'backbone'
|
28 |
+
lr: 0.000001
|
configs/rtdetr/rtdetr_r101vd_6x_cococope24.yml
ADDED
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
__include__: [
|
2 |
+
'../dataset/coco_detection.yml',
|
3 |
+
'../runtime.yml',
|
4 |
+
'./include/dataloader.yml',
|
5 |
+
'./include/optimizer.yml',
|
6 |
+
'./include/rtdetr_r50vdcope24.yml',
|
7 |
+
]
|
8 |
+
|
9 |
+
PResNet:
|
10 |
+
depth: 101
|
11 |
+
|
12 |
+
|
13 |
+
HybridEncoder:
|
14 |
+
# intra
|
15 |
+
hidden_dim: 384
|
16 |
+
dim_feedforward: 2048
|
17 |
+
|
18 |
+
|
19 |
+
RTDETRTransformer:
|
20 |
+
feat_channels: [384, 384, 384]
|
21 |
+
|
22 |
+
|
23 |
+
optimizer:
|
24 |
+
type: AdamW
|
25 |
+
params:
|
26 |
+
-
|
27 |
+
params: 'backbone'
|
28 |
+
lr: 0.000001
|