针对mnn模型处理,输出增加shape
This commit is contained in:
@ -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) {
|
||||
|
Reference in New Issue
Block a user