姿势识别源代码matlab
关键点检测
我们的模型使用带有递归模块的卷积神经网络从单个图像检测身体关键点。
下面,我们提供代码来训练和测试模型。
有关基于Web的演示,请访问。
软件依赖关系
您需要MATLAB和工具箱来运行此演示。
该演示已针对Matlab
2016a,CUDNN
v5.1和cuda
8.0(使用Linux机器)进行了编译和测试。
训练模型(MPII人类姿势数据集)
通过执行getMPIIData_v3.m然后拆分MPIIData_V4.m下载并准备数据集。
数据准备需要几分钟才能完成。
输出是火车(
MPI_imdbsT1aug0.mat
,
〜3GB
)和验证(
MPI_imdbsV1aug0.mat
,〜0.3GB)文件。
执行trainBodyPose_example.m
(需要设置一些参数,例如MatConvNet路径和GPU索引)。
要测试模型,请运行下面的演示,如下所述。
用一次循环迭代训练模型的参数:
net
=
initialize3ObjeRecFusion(opts,2,0,'shareFlag',[0,1]);
opts.derOutputs
2022-06-06 11:31:09
413KB
系统开源
1