lyffly / KeyPointsDetection

using deep learining to detect keypoints in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

KeyPointsDetection 关键点识别网络

程序编写与测试:刘云飞

建议与合作联系邮箱:liuyunfei.1314@163.com

0x00 语言和工具

框架:PyTorch 1.2.0

语言:Python 3.7

在ubuntu 18.03测试通过

0x01 数据准备

标注数据类似下面,512x512 pixel

图片1

图片2

图片2

增加上2D高斯 点云 的Heatmap图如下

热点图

0x02 训练

训练使用Houeglass模型,128*128进行训练

基本结构如下,漏斗式结构

网络结构图

0x03 结果

下图左侧为检测到的四个点的heatmap,右侧为加上原图的效果,可以看到效果还不错~

结果图例1

结果1图

结果图例2

结果2图

0x04 后续

人体关键点检测

人脸关键点检测

About

using deep learining to detect keypoints in PyTorch


Languages

Language:Python 100.0%