function preprocess(imageData) {
const {
width,
height,
data
} = imageData;
const dataTensor = ndarray(new Float32Array(data), [width, height, 4])
const dataProcessedTensor = ndarray(new Float32Array(width * height * 3), [width, height, 3])
//从[0,255]转化到[0,1]
ops.divseq(dataTensor, 255)
//获取R数据
ops.assign(dataProcessedTensor.pick(null, null, 0), dataTensor.pick(null, null, 0))
//获取G数据
ops.assign(dataProcessedTensor.pick(null, null, 1), dataTensor.pick(null, null, 1))
//获取B数据
ops.assign(dataProcessedTensor.pick(null, null, 2), dataTensor.pick(null, null, 2))
const preprocessedData = dataProcessedTensor.data;
console.log('via process');
return preprocessedData;
}
找不到 ndarray 和 ops 请问我该如何引用 ?谢谢 谢谢
添加回答
举报
0/150
提交
取消