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

2025-03-10 17:49:53 +08:00
parent e362890c96
commit a966b82963
13 changed files with 51 additions and 29 deletions
--- a/src/deploy/faceattr/gender-age.ts
+++ b/src/deploy/faceattr/gender-age.ts
@ -36,7 +36,7 @@ export class GenderAge extends Model {
 				data: nchwImage,
 				type: "float32",
 			}
-		}).then(res => res[output.name]);
+		}).then(res => res[output.name].data);

 		return {
 			gender: result[0] > result[1] ? "F" : "M",