ready for merge

Author: fbelderink

android/src/main/java/io/fynn/pytorch_mobile/PyTorchMobilePlugin.java

@@ -116,8 +116,6 @@ public void onMethodCall(@NonNull MethodCall call, @NonNull Result result) {
           // Custom std
           ArrayList<Double> _std = call.argument("std");
           std = Convert.toFloatPrimitives(_std.toArray(new Double[0]));
-          
-          
 
           imageModule = modules.get(index);
 

lib/model.dart

@@ -31,8 +31,9 @@ class Model {
       {List<double> mean = TORCHVISION_NORM_MEAN_RGB,
       List<double> std = TORCHVISION_NORM_STD_RGB}) async {
     // Assert mean std
-    assert(mean.length == 3, "Mean should have size of 3");
-    assert(std.length == 3, "STD should have size of 3");
+    assert(mean.length == 3, "mean should have size of 3");
+    assert(std.length == 3, "std should have size of 3");
+
     List<String> labels = await _getLabels(labelPath);
     List byteArray = image.readAsBytesSync();
     final List? prediction = await _channel.invokeListMethod("predictImage", {