Update script_B7.py
Browse filesChange 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])])
- 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 |
-
|
49 |
-
|
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.
|