"""下面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()会用训练时固定的均值方差参数来推理,不要搞错
"""
相关文章
- 02-01pose问题:连续帧的pose一直在飘