直接回归的整个流程如下:
- 用卷积提特征
- 用全连接层进行回归得到关键点
- 与标签做损失并优化损失
当图像很大的时候,神经网络是在整张图像中去寻找一个像素点,所以直接回归这一解决办法让神经网络的工作是非常难的。在下一篇我们会手撕热力图方案的代码。热力图就是在关键点周围生成一个高斯圆,整个圆都可以看作是当前的关键点,但是整个圆是有一个概率值在里面,最中心的概率值最大,对应的热力图也越红,越往外概率越小。这种方法相对直接回归来说正样本更多,学起来也就更快,效果也要比直接回归好很多。首先我们需要一些pytorch的基本知识,大佬可以直接略过~
数据加载器前置学习
训练模型一般都是先解决数据的输入问题和预处理问题