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
|