add model
Browse files
rtmdet_m_640-8xb32_coco-person.py
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_base_ = "mmdet::rtmdet/rtmdet_m_8xb32-300e_coco.py"
|
2 |
+
|
3 |
+
checkpoint = "https://download.openmmlab.com/mmdetection/v3.0/rtmdet/cspnext_rsb_pretrain/cspnext-m_8xb256-rsb-a1-600e_in1k-ecb3bbd9.pth"
|
4 |
+
|
5 |
+
model = {
|
6 |
+
"backbone": {
|
7 |
+
"init_cfg": {"type": "Pretrained", "prefix": "backbone.", "checkpoint": checkpoint},
|
8 |
+
},
|
9 |
+
"bbox_head": {"num_classes": 1},
|
10 |
+
"test_cfg": {
|
11 |
+
"nms_pre": 1000,
|
12 |
+
"min_bbox_size": 0,
|
13 |
+
"score_thr": 0.05,
|
14 |
+
"nms": {"type": "nms", "iou_threshold": 0.6},
|
15 |
+
"max_per_img": 100,
|
16 |
+
},
|
17 |
+
}
|
18 |
+
|
19 |
+
train_dataloader = {"dataset": {"metainfo": {"classes": ("person",)}}}
|
20 |
+
|
21 |
+
val_dataloader = {"dataset": {"metainfo": {"classes": ("person",)}}}
|
22 |
+
test_dataloader = val_dataloader
|
rtmdet_m_8xb32-100e_coco-obj365-person-235e8209.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:35b0c7406499e0d141dd6a0235db07c10d2bee8f891f8f4e353c16a009de30e8
|
3 |
+
size 99013732
|
rtmpose-l_simcc-coco-wholebody_pt-aic-coco_270e-384x288-eaeb96c8_20230125.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:13ce77ad08808333e2d4f850c632ac6068c837108e15f1c3902dcbcf30842db9
|
3 |
+
size 134731115
|