1 回答

TA貢獻1848條經驗 獲得超2個贊
你的問題有幾個問題。
首先color_map只有 5 個元素,但預期結果的索引從 0 到 5(6 個元素),我認為這是一個錯誤,你只需要真正的索引。
其次,您的代碼中沒有index分配的值,所以我假設它是下一個可用索引,并改用push屬性。
由于您實際上不想返回多維數組,而只想返回二維索引數組,因此返回imageData.
考慮到您在評論部分解釋的條件,即顏色映射值將是您唯一可以嘗試做的事情:
const color_map = [[255,255,0], [255,0,0], [0,255,255], [0,255,0], [0,0,0]];
function form_2D_label(mat) {
const image = mat.cvtColor(cv.COLOR_BGR2RGB);
const imageBuffer = mat.getData();
const ui8 = new Uint8Array(imageBuffer);
const imageData = [];
for (let i = 0; i < ui8.length; i += 3) {
imageData.push([ui8[i], ui8[i + 1], ui8[i + 2]]);
console.log(imageData[imageData.length - 1])
}
return [imageData.map(el => color_map.findIndex(color => arrayEquals(color, el)))];
}
function arrayEquals(array1, array2) {
for (let i = 0, l = array2.length; i < l; i++) {
if (array2[i] !== array1[i]) return false;
}
return true;
}
添加回答
舉報