chirmy commited on
Commit
a25fba5
·
verified ·
1 Parent(s): c04e21a

Update script_B7.py

Browse files

Change T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])]) to T.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])

Files changed (1) hide show
  1. script_B7.py +3 -3
script_B7.py CHANGED
@@ -45,9 +45,9 @@ class PytorchWorker:
45
 
46
  self.transforms = T.Compose([T.Resize((299, 299)),
47
  T.ToTensor(),
48
- T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])
49
- # T.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])
50
-
51
 
52
  def predict_image(self, image: np.ndarray) -> list():
53
  """Run inference using ONNX runtime.
 
45
 
46
  self.transforms = T.Compose([T.Resize((299, 299)),
47
  T.ToTensor(),
48
+ T.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])
49
+ # T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])])
50
+
51
 
52
  def predict_image(self, image: np.ndarray) -> list():
53
  """Run inference using ONNX runtime.