From b8119e65206d74728644559d2912fece2f8b4094 Mon Sep 17 00:00:00 2001
From: Lucas Oskorep <lucas.oskorep@gmail.com>
Date: Sat, 2 Jul 2022 00:34:11 -0400
Subject: [PATCH] fixing pre-processing pipeline

---
 lib/tflite/classifier.dart | 9 +++++----
 lib/tflite/ml_isolate.dart | 1 -
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/lib/tflite/classifier.dart b/lib/tflite/classifier.dart
index bc3bf4c..0a44685 100644
--- a/lib/tflite/classifier.dart
+++ b/lib/tflite/classifier.dart
@@ -30,7 +30,6 @@ class Classifier {
   late List<String> _labels;
   int classifierCreationStart = -1;
 
-
   Classifier({
     Interpreter? interpreter,
     List<String>? labels,
@@ -78,6 +77,7 @@ class Classifier {
       imageProcessor ??= ImageProcessorBuilder()
           .add(ResizeWithCropOrPadOp(224, 224))
           .add(ResizeOp(inputSize, inputSize, ResizeMethod.BILINEAR))
+          .add(NormalizeOp(0, 1))
           // .add(NormalizeOp(127.5, 127.5))
           .build();
       return imageProcessor?.process(inputImage);
@@ -93,9 +93,9 @@ class Classifier {
     var inferenceStart = DateTime.now().millisecondsSinceEpoch;
     _interpreter.run(_inputImage.buffer, _outputBuffer.getBuffer());
     var postProcStart = DateTime.now().millisecondsSinceEpoch;
-    Map<String, double> labeledProb = TensorLabel.fromList(
-            labels, _outputProcessor.process(_outputBuffer))
-        .getMapWithFloatValue();
+    Map<String, double> labeledProb =
+        TensorLabel.fromList(labels, _outputProcessor.process(_outputBuffer))
+            .getMapWithFloatValue();
     final predictions = getTopProbabilities(labeledProb, number: 5)
         .mapIndexed(
             (index, element) => Recognition(index, element.key, element.value))
@@ -111,6 +111,7 @@ class Classifier {
       ),
     };
   }
+
   /// Gets the interpreter instance
   Interpreter get interpreter => _interpreter;
 
diff --git a/lib/tflite/ml_isolate.dart b/lib/tflite/ml_isolate.dart
index 4ea45a2..4e9314f 100644
--- a/lib/tflite/ml_isolate.dart
+++ b/lib/tflite/ml_isolate.dart
@@ -5,7 +5,6 @@ import 'package:tensordex_mobile/tflite/classifier.dart';
 import 'package:tflite_flutter/tflite_flutter.dart';
 
 import '../utils/image_utils.dart';
-import '../utils/logger.dart';
 
 class IsolateBase {
   final ReceivePort _receivePort = ReceivePort();