Skip to content

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

good sample to no defect

parent 0699d4fd
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
scripts/welspun_classifier.py
View file @ 56cc7edc
...@@ -178,7 +178,7 @@ class Welspun_Classifier(ModelWrapper): ...@@ -178,7 +178,7 @@ 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
......
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