fromPixel根据平台产生不同的张量

时间:2018-12-08 17:15:02

标签: tensorflow.js

我正在使用此代码段将图像加载到张量中

var img = document.getElementById("myImage");  
var example =tf.fromPixels(img);
example = tf.image.resizeBilinear(example, [150, 150], align_corners=true);
example = example.toFloat().div(tf.scalar(255));
example = example.reshape([1, 150, 150, 3]); 
example.print()

在tensorflow.js中使用某些模型对图像进行分类时,我意识到根据运行位置(例如笔记本电脑与Pixel 3手机),我获得的softmax输出会略有不同。

进一步调试,我意识到从tf.fromPixels()获得的张量值根据平台而略有不同。

这是可以预期的吗?任何人都知道为什么会发生这种情况以及如何防止这种情况发生?

1 个答案:

答案 0 :(得分:3)

tf.fromPixels使用HTMLCanvasElement。浏览器倾向于使用不同的图像处理引擎等。因此,类似的canvas元素可以根据系统产生不同的值。

更多信息: