commit a1316f0b425607a0ed6c0383efbb91d9aac155e3
author Lun Dong <lundong@google.com> Wed Aug 09 10:08:46 2023 -0700
committer Lun Dong <lundong@google.com> Wed Aug 09 10:08:46 2023 -0700
tree 8979ee7a5e3aefc6efc9017a281dc776666de97a
parent 9c58c88cc07da691339a4171fc0cc61048ecf2fe

ml/ml-models-public: update stored output to match printed log

Change-Id: Ibc5b68a9042a4b9e5bc19a6fb57cbd08f7da065b

float_models/iree_exec/mobilenet_v1.c

diff --git a/float_models/iree_exec/mobilenet_v1.c b/float_models/iree_exec/mobilenet_v1.c
index e8cb740..7094746 100644
--- a/float_models/iree_exec/mobilenet_v1.c
+++ b/float_models/iree_exec/mobilenet_v1.c
@@ -75,7 +75,7 @@
     }
   }
   score.best_out = best_out;
-  score.best_idx = best_idx;
+  score.best_idx = best_idx + 1;
 
   LOG_INFO("Image prediction result is: id: %d", best_idx + 1);
 

quant_models/iree_exec/mobilenet_v1.c

diff --git a/quant_models/iree_exec/mobilenet_v1.c b/quant_models/iree_exec/mobilenet_v1.c
index 4c44dbb..501204f 100644
--- a/quant_models/iree_exec/mobilenet_v1.c
+++ b/quant_models/iree_exec/mobilenet_v1.c
@@ -76,7 +76,7 @@
     }
   }
   score.best_out = best_out;
-  score.best_idx = best_idx;
+  score.best_idx = best_idx + 1;
 
   LOG_INFO("Image prediction result is: id: %d", best_idx + 1);
 

quant_models/iree_exec/mobilenet_v2.c

diff --git a/quant_models/iree_exec/mobilenet_v2.c b/quant_models/iree_exec/mobilenet_v2.c
index c625520..72d36eb 100644
--- a/quant_models/iree_exec/mobilenet_v2.c
+++ b/quant_models/iree_exec/mobilenet_v2.c
@@ -76,7 +76,7 @@
     }
   }
   score.best_out = best_out;
-  score.best_idx = best_idx;
+  score.best_idx = best_idx + 1;
 
   LOG_INFO("Image prediction result is: id: %d", best_idx + 1);