未验证 提交 be40be8a 编写于 作者: K KP 提交者: GitHub

Merge pull request #1879 from rainyfly/add_pp_tinypose_module

update pp_tinypose
......@@ -3,7 +3,7 @@
|模型名称|pp-tinypose|
| :--- | :---: |
|类别|图像-关键点检测|
|网络|PicoDet + HRNet|
|网络|PicoDet + tinypose|
|数据集|COCO + AI Challenger|
|是否支持Fine-tuning|否|
|模型大小|125M|
......@@ -20,7 +20,7 @@
<br />
输入图像
<br />
<img src="https://user-images.githubusercontent.com/22424850/169768604-d23a1851-c18b-4f9f-a8ab-2c3f3080e393.jpg" hspace='10'/>
<img src="https://user-images.githubusercontent.com/22424850/170029768-3c60def2-7c87-4e8a-98bc-1bbc912204e7.jpg" hspace='10'/>
<br />
输出图像
......
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册