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

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

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

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

数据加载器前置学习

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

Original: https://blog.csdn.net/qq_42308217/article/details/124825075
Author: @会飞的毛毛虫
Title: 关键点检测之直接回归(逐行手撕Pytorch)

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/710342/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球