diff --git a/modules/image/keypoint_detection/pp-tinypose/README.md b/modules/image/keypoint_detection/pp-tinypose/README.md index 7c11c495c4b0b65695d71e35d2254281d879bdbb..caa24745018291271851af1f46487eab0bda376d 100644 --- a/modules/image/keypoint_detection/pp-tinypose/README.md +++ b/modules/image/keypoint_detection/pp-tinypose/README.md @@ -3,7 +3,7 @@ |模型名称|pp-tinypose| | :--- | :---: | |类别|图像-关键点检测| -|网络|PicoDet + HRNet| +|网络|PicoDet + tinypose| |数据集|COCO + AI Challenger| |是否支持Fine-tuning|否| |模型大小|125M| @@ -20,7 +20,7 @@
输入图像
- +
输出图像 diff --git a/modules/image/keypoint_detection/pp-tinypose/module.py b/modules/image/keypoint_detection/pp-tinypose/module.py index 4d1efc82f65e717c6deb67c50ca127db7cfc87fa..c0aa6fd88ec77ab8ccd36c6f8d192a269e9e293f 100644 --- a/modules/image/keypoint_detection/pp-tinypose/module.py +++ b/modules/image/keypoint_detection/pp-tinypose/module.py @@ -48,7 +48,7 @@ class PP_TinyPose: def __init__(self): self.det_model_dir = os.path.join(self.directory, 'model/picodet_s_320_coco_lcnet/') - self.keypoint_model_dir = os.path.join(self.directory, 'model/dark_hrnet_w32_256x192/') + self.keypoint_model_dir = os.path.join(self.directory, 'model/tinypose_256x192/') self.detector = Detector(self.det_model_dir) self.topdown_keypoint_detector = KeyPointDetector(self.keypoint_model_dir)