针对mnn模型处理,输出增加shape

This commit is contained in:
2025-03-10 17:49:53 +08:00
parent e362890c96
commit a966b82963
13 changed files with 51 additions and 29 deletions

View File

@ -51,7 +51,7 @@ export class PFLD extends Model {
shape: [1, 3, input.shape[2], input.shape[3]],
}
});
const pointsBuffer = res[pointsOutput.name];
const pointsBuffer = res[pointsOutput.name].data;
const points: FacePoint[] = [];
for (let i = 0; i < pointsBuffer.length; i += 2) {