Javascript：如何将 opencv mat 转换为张量？

Question

有人知道如何将 opencv.js mat 转换为张量，以便我可以将其输入到我的 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])