Spaces:
Runtime error
Runtime error
File size: 1,195 Bytes
cc0dd3c |
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 |
# 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='rtmdet-m', weights="/fsx_laion/alvin/pretrain/ViTPose/rtmdet_m_8xb32-300e_coco_20220719_112220-229f527c.pth", cat_ids=(0, )),
face=dict(
model=osp.join(mmpose_path, '.mim',
'demo/mmdetection_cfg/yolox-s_8xb8-300e_coco-face.py'),
weights='/fsx_laion/alvin/pretrain/ViTPose/yolo-x_8xb8-300e_coco-face_13274d7c.pth',
cat_ids=(0, )),
hand=dict(
model=osp.join(
mmpose_path, '.mim', 'demo/mmdetection_cfg/'
'ssdlite_mobilenetv2_scratch_600e_onehand.py'),
weights='/fsx_laion/alvin/pretrain/ViTPose/ssdlite_mobilenetv2_scratch_600e_onehand-4f9f8686_20220523.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']
|