Merge pull request NVIDIA#149 from HanbumKo/HanbumKo-patch-1

skierat · web-flow · commit 2fb7c8f472ff · 2019-12-09T11:32:36.000+01:00
Update inference.py
diff --git a/PyTorch/Detection/SSD/dle/inference.py b/PyTorch/Detection/SSD/dle/inference.py
@@ -17,7 +17,6 @@
 
 def load_image(image_path):
     """Code from Loading_Pretrained_Models.ipynb - a Caffe2 tutorial"""
-    mean, std = 128, 128
     img = skimage.img_as_float(skimage.io.imread(image_path))
     if len(img.shape) == 2:
         img = np.array([img, img, img]).swapaxes(0,2)
