ml/ml-models-public: update stored output to match printed log Change-Id: Ibc5b68a9042a4b9e5bc19a6fb57cbd08f7da065b
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);
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);
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);