ycdhqzhiai / Gaze-PFLD

PFLD实现Gaze Estimation
20 stars 6 forks source link

数据归一化 #6

Closed fengyanWang closed 2 years ago

fengyanWang commented 2 years ago

你好,我看你推理部分代码没有对读入图片做数据的归一化,请问这是正常的吗?还有开源的模型效果是readme中那张图吗?期待回复,谢谢!

ycdhqzhiai commented 2 years ago

https://github.com/ycdhqzhiai/Gaze-PFLD/blob/2dcd284943ee34e84373b5b5215bf8b5f56fea7c/test.py#L128 这个0-1就是归一化啊

fengyanWang commented 2 years ago

这是转pytorch的tensor吧。我想知道的是减去均值除方差这种操作

ycdhqzhiai commented 2 years ago

你需要仔细读下代码,看看API