ml/ml-models-public: use intermediate output for person_detection model Change test output range accordingly. Change-Id: Ieab7d8f8683721ec81903973ad9590d6d56b7289
diff --git a/quant_models/CMakeLists.txt b/quant_models/CMakeLists.txt index e8dec30..b6fef98 100644 --- a/quant_models/CMakeLists.txt +++ b/quant_models/CMakeLists.txt
@@ -52,6 +52,9 @@ "person_detection.tflite" C_IDENTIFIER "quant_models_person_detection" + TFLITE_OUTPUT + # TODO(b/287490866): remove this argument after IREE fix is landed + "--output-array=MobilenetV1/Logits/SpatialSqueeze" FLAGS "-iree-input-type=tosa" "-riscv-v-fixed-length-vector-lmul-max=8"
diff --git a/quant_models/iree_exec/person_detection_test.filecheck b/quant_models/iree_exec/person_detection_test.filecheck index 1a880de..557f3bf 100644 --- a/quant_models/iree_exec/person_detection_test.filecheck +++ b/quant_models/iree_exec/person_detection_test.filecheck
@@ -1 +1,2 @@ -// CHECK: {{Non-person Score: -11[3-4]; Person Score: 11[3-4]}} +// TODO(b/287490866): change range after IREE fix is landed +// CHECK: {{Non-person Score: -11[2-4]; Person Score: 11[0-4]}}