openvino

3b0a1bb7 · dasharatha.vamshi · 61ca2cce · 3b0a1bb7
Commit 3b0a1bb7 authored Feb 12, 2021 by dasharatha.vamshi
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

scripts/welspun_classifier.py scripts/welspun_classifier.py +2 -2

No files found.
--- a/scripts/welspun_classifier.py
+++ b/scripts/welspun_classifier.py
@@ -178,8 +178,8 @@ class Welspun_Classifier(ModelWrapper):
        # imagenet_stats
        mean = [0.485, 0.456, 0.406]
        std = [0.229, 0.224, 0.225]
-        # image = image * 1 / 255.0
-        # image = (image - mean) / std
+        image = image * 1 / 255.0
+        image = (image - mean) / std
        return image

    def softmax_function(self, vec):