Skip to content

W
Welspun-Classification
Commit 3b0a1bb7 authored by dasharatha.vamshi's avatar dasharatha.vamshi
Browse files
Options

openvino

parent 61ca2cce
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
scripts/welspun_classifier.py
View file @ 3b0a1bb7
...@@ -178,8 +178,8 @@ class Welspun_Classifier(ModelWrapper): ...@@ -178,8 +178,8 @@ class Welspun_Classifier(ModelWrapper):
# imagenet_stats # imagenet_stats
mean = [0.485, 0.456, 0.406] mean = [0.485, 0.456, 0.406]
std = [0.229, 0.224, 0.225] std = [0.229, 0.224, 0.225]
# image = image * 1 / 255.0 image = image * 1 / 255.0
# image = (image - mean) / std image = (image - mean) / std
return image return image
def softmax_function(self, vec): def softmax_function(self, vec):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment