From 201649595e659dea36b527358b94b3a0c52b4fff Mon Sep 17 00:00:00 2001 From: "Eric.Lee2021" <305141918@qq.com> Date: Sat, 19 Jun 2021 20:04:31 +0800 Subject: [PATCH] update mode & usage --- README.md | 35 ++++++++++++++++++++++++++++++++--- yolo_inference.py | 4 ++-- 2 files changed, 34 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 4734cfc..f7e1c0b 100644 --- a/README.md +++ b/README.md @@ -33,17 +33,46 @@ Easy 3D HandPose,pytorch,单目相机的手三维姿态估计 ### 2、预训练模型 -* [预训练模型下载地址(百度网盘 Password: 后续更新 )]() +* [预训练模型下载地址(百度网盘 Password: 95t4 )](https://pan.baidu.com/s/1L9JVjnvKDjG0opIAUZOF0g) ## 项目使用方法 +### 准备左右手3D建模资源(MANO Hand Model) +* 该模型也可在官网下载,官网地址为:https://mano.is.tue.mpg.de/ +* 下载模型和文件 (下载文件的格式为 mano_v*_*.zip)。注意这些文件的下载使用遵守 [MANO license](https://mano.is.tue.mpg.de/license)。 +* 下载模型后进行解压,并将其目录结构设置如下: +``` +e3d_handpose_x/ + mano/ + models/ + MANO_LEFT.pkl + MANO_RIGHT.pkl + ... + webuser/ + ... + __init__.py +``` ### 模型训练 * 根目录下运行命令: python train.py (注意脚本内相关参数配置 ) ### 模型推理 * 根目录下运行命令: -#### python inference.py (注意脚本内相关参数配置 ) -#### python yolo_inference.py (注意脚本内相关参数配置 ) +* 图片推理脚本 :python inference.py (注意脚本内相关参数配置 ) + +* 相机推理脚本 : python yolo_inference.py (注意脚本内相关参数配置 ) +#### 第 1 步:确定电脑连接相机。 +#### 第 2 步:下载[模型前向推理包(百度网盘 Password: xhd3 )](https://pan.baidu.com/s/1wqhIgciL5mnlT1PyHKI6QQ) +#### 第 3 步:解压模型前向推理包,配置 [yolo_inference.py](https://codechina.csdn.net/EricLee/e3d_handpose_x/-/blob/master/yolo_inference.py)脚本模型路径参数,参考如下: +``` + parser.add_argument('--model_path', type=str, default = './if_package/e3d_handposex-resnet_50-size-128-loss-wing_loss-20210619', + help = 'model_path') # e3d handpose 模型路径 + parser.add_argument('--detect_model_path', type=str, default = './if_package/hand_detect_416-20210606.pt', + help = 'model_path') # detect 模型路径 + parser.add_argument('--handpose_x2d_model_path', type=str, default = './if_package/handposex_2d_resnet_50-size-256-wingloss102-0.119.pth', + help = 'model_path') # 手2维关键点 模型路径 +``` +#### 第 4 步:运行脚本:python yolo_inference.py +#### 注意:运行出错,注意看log报的错误,尽量自行解决,思考尝试解决不了,issue提问。 ## 联系方式 (Contact) * E-mails: 305141918@qq.com diff --git a/yolo_inference.py b/yolo_inference.py index 78e6929..468c94c 100644 --- a/yolo_inference.py +++ b/yolo_inference.py @@ -1,7 +1,7 @@ #-*-coding:utf-8-*- # date:2021-06-15 # Author: Eric.Lee -# function: handpose 3D Inference +# function: handpose 3D Yolo_v3 Detect Inference import os import argparse @@ -36,7 +36,7 @@ from manopth import manolayer if __name__ == "__main__": parser = argparse.ArgumentParser(description=' Project Hand Pose 3D Inference') - parser.add_argument('--model_path', type=str, default = '././model_exp/2021-06-19_14-20-56/resnet_50-size-128-loss-wing_loss-model_epoch-196.pth', + parser.add_argument('--model_path', type=str, default = './if_package/e3d_handposex-resnet_50-size-128-loss-wing_loss-20210619.pth', help = 'model_path') # e3d handpose 模型路径 parser.add_argument('--detect_model_path', type=str, default = './if_package/hand_detect_416-20210606.pt', help = 'model_path') # detect 模型路径 -- GitLab