Replace squeeze with reshape for facenet_nchw.js (webmachinelearning#194

)
Honry · May 15, 2024 · 064c4c4 · 064c4c4
1 parent 5d98ae3
commit 064c4c4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/face_recognition/facenet_nchw.js b/face_recognition/facenet_nchw.js
@@ -233,7 +233,10 @@ export class FaceNetNchw {
         block8_5, 6, ['977', '1104', '978', '1080', '1086'], false);
 
     const averagePool = this.builder_.averagePool2d(block8_6);
-    const squeeze = this.builder_.squeeze(averagePool, {axes: [2, 3]});
+    // Use reshape to implement squeeze(averagePool, {axes: [2, 3]});
+    const squeezed_shape = averagePool.shape();
+    squeezed_shape.splice(2, 2);
+    const squeeze = this.builder_.reshape(averagePool, squeezed_shape);
     const gemm = await this.buildGemm_(squeeze);
     // L2Normalization will be handled in post-processing
     return gemm;