Javascript:如何将opencv mat转换为tensor？

Question

有谁知道如何将opencv.js垫子转换为张量，这样我就可以将其输入到我的tensorflow.js分类器中？

以下代码显示了我在读取和预处理要分类的图像时所做的工作：

img_array = cv.imread(document.getElementById('picture1'), cv.IMREAD_GRAYSCALE);
cv.cvtColor(img_array, img_array, cv.COLOR_RGBA2GRAY);
let dsize = new cv.Size(100, 100);
cv.resize(img_array, img_array, dsize);

我的分类器需要一个形状为(1,100,100，1)的张量作为输入，而我不知道如何将cv mat转换为tensorflow.js张量。

Answer 1

图像对象具有属性data，可用于获取平面化数组中的所有像素值。要构造张量，可以使用以下内容

const src = cv.imread(imageSource)

const tensor = tf.tensor(src.data, [src.rows, src.cols, -1])