update

w-okada · May 13, 2022 · 201bba5 · 201bba5
1 parent 6e803f7
commit 201bba5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tfl009_face_hand_pose_mix/resources/wasm/tflite-simd.wasm b/tfl009_face_hand_pose_mix/resources/wasm/tflite-simd.wasm
diff --git a/tfl009_face_hand_pose_mix/wasm/hand-core.hpp b/tfl009_face_hand_pose_mix/wasm/hand-core.hpp
@@ -220,15 +220,15 @@ class HandCore
         {
             int tensor_idx = handLandmarkInterpreter->outputs()[j];
             const char *tensor_name = handLandmarkInterpreter->tensor(tensor_idx)->name;
-            if (strcmp(tensor_name, "ld_21_3d") == 0 || strcmp(tensor_name, "Identity") == 0)
+            if (strcmp(tensor_name, "ld_21_3d") == 0 || strcmp(tensor_name, "Identity") == 0 || strcmp(tensor_name, "Identity:0") == 0)
             {
                 landmark_ptr = handLandmarkInterpreter->typed_output_tensor<float>(j);
             }
-            else if (strcmp(tensor_name, "output_handflag") == 0 || strcmp(tensor_name, "Identity_1") == 0)
+            else if (strcmp(tensor_name, "output_handflag") == 0 || strcmp(tensor_name, "Identity_1") == 0 || strcmp(tensor_name, "Identity_1:0") == 0)
             {
                 handflag_ptr = handLandmarkInterpreter->typed_output_tensor<float>(j);
             }
-            else if (strcmp(tensor_name, "Identity_2") == 0)
+            else if (strcmp(tensor_name, "Identity_2") == 0 || strcmp(tensor_name, "Identity_2:0") == 0)
             {
                 handedness_ptr = handLandmarkInterpreter->typed_output_tensor<float>(j);
             }