关键点检测之直接回归(逐行手撕Pytorch)

直接回归的整个流程如下:

  • 用卷积提特征
  • 用全连接层进行回归得到关键点
  • 与标签做损失并优化损失

当图像很大的时候,神经网络是在整张图像中去寻找一个像素点,所以直接回归这一解决办法让神经网络的工作是非常难的。在下一篇我们会手撕热力图方案的代码。热力图就是在关键点周围生成一个高斯圆,整个圆都可以看作是当前的关键点,但是整个圆是有一个概率值在里面,最中心的概率值最大,对应的热力图也越红,越往外概率越小。这种方法相对直接回归来说正样本更多,学起来也就更快,效果也要比直接回归好很多。首先我们需要一些pytorch的基本知识,大佬可以直接略过~


数据加载器前置学习

        训练模型一般都是先解决数据的输入问题预处理问题

猜你喜欢

转载自blog.csdn.net/qq_42308217/article/details/124825075