pose问题:连续帧的pose一直在飘

  """下面2行注释掉"""
        self.det_model.to(device)
        # self.det_model.train()
        self.pose_model.to(device)
        # self.pose_model.train()
        self.model.to(device)
        self.model.train()

"""
 对视频提取pose结果实时可视化显示,连续帧的pose一直在飘,推理时 输入帧间变化很小,理论上pose结果应该很稳定才对.
因此推测问题原因是帧间推理时,权重文件一直在变化,即模型没有处于eval()模式
.train是训练模式,不会固定BN层参数,根据每次迭代计算得到BN的均值和方差;而.eval()会用训练时固定的均值方差参数来推理,不要搞错
"""
posted @ 2020-11-23 13:52  Parallax  阅读(125)  评论(0编辑  收藏  举报