/CPM_landmark

基于CPM网络结构加入moblienet**的关键点检测模型

Primary LanguagePython

CPM_landmark

基于CPM网络结构的关键点检测模型

版本要求

py2 or py3
tensorflow>=1.2

opencv>=3.2

文件说明

bt.py 生成带有heatmap的训练数据的tfrecords
bv.py 生成带有heatmap的验证数据的tfrecords

blouse.py 训练模型
blouse_test.py 测试数据

visual.py 可视化

参考文献

Wei S E, Ramakrishna V, Kanade T, et al. Convolutional Pose Machines[J]. 2016:4724-4732.
Howard A G, Zhu M, Chen B, et al. MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications[J]. 2017.