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

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 矩阵转换为 tensorflow.js 张量。

Answer 1

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

const src = cv.imread(imageSource)

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

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

Javascript: How to convert opencv mat to tensor?

javascript

tensorflow

opencv

image-processing

tensorflow.js