Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,391 Bytes
a249588 |
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 |
# Copyright (c) OpenMMLab. All rights reserved.
import os.path as osp
from mmengine.config.utils import MODULE2PACKAGE
from mmengine.utils import get_installed_path
mmpose_path = get_installed_path(MODULE2PACKAGE['mmpose'])
default_det_models = dict(
human=dict(
model=osp.join(
mmpose_path, '.mim', 'demo/mmdetection_cfg/'
'rtmdet_m_640-8xb32_coco-person.py'),
weights='https://download.openmmlab.com/mmpose/v1/projects/'
'rtmposev1/rtmdet_m_8xb32-100e_coco-obj365-person-235e8209.pth',
cat_ids=(0, )),
face=dict(
model=osp.join(mmpose_path, '.mim',
'demo/mmdetection_cfg/yolox-s_8xb8-300e_coco-face.py'),
weights='https://download.openmmlab.com/mmpose/mmdet_pretrained/'
'yolo-x_8xb8-300e_coco-face_13274d7c.pth',
cat_ids=(0, )),
hand=dict(
model=osp.join(mmpose_path, '.mim', 'demo/mmdetection_cfg/'
'rtmdet_nano_320-8xb32_hand.py'),
weights='https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/'
'rtmdet_nano_8xb32-300e_hand-267f9c8f.pth',
cat_ids=(0, )),
animal=dict(
model='rtmdet-m',
weights=None,
cat_ids=(15, 16, 17, 18, 19, 20, 21, 22, 23)),
)
default_det_models['body'] = default_det_models['human']
default_det_models['wholebody'] = default_det_models['human']
|