blazepose的文件下载
链接:https://pan.baidu.com/s/1bc_5VBdpG-bIU8NQqnW0hA
提取码:本人QQ 3233426794
一、制作数据集
这里使用的lsp数据集,数据集的下载地址为:
链接:https://pan.baidu.com/s/1mIG2E9mRRk1DCkU2SYbaNg?pwd=0310
lsp数据集包括:
二、训练
1.热图训练
将 config.py 文件中的两个位置设为: train_mode=0,continue_mode=0 。设置完成之后运行:python train.py即可
运行效果为:
2.回归训练
将 config.py 文件中的两个位置设为: train_mode=1,continue_mode=0 。设置完成之后运行:python train.py即可
注意:热图训练完成之后才能进行回归训练
三、测试
将 config.py 文件中的两个位置设为: train_mode=1,eval_mode=0 。设置完成之后运行:python test.py即可
四、预测