Warning no more

MobileNetV1.py

@@ -11,6 +11,8 @@ import numpy as np
 
 from coco_utils import get_image
 
+import warnings
+warnings.filterwarnings("ignore", category=UserWarning) 
 
 model = torch.hub.load('pytorch/vision:v0.10.0', 'mobilenet_v2', pretrained=True)
 #Enable following to only get CNN part:
@@ -26,7 +28,7 @@ preprocess = transforms.Compose([
 
 #Outputs mini-batch that can be input to model
 def preprocessImage(image):
-  input_tensor = preprocess(input_image)
+  input_tensor = preprocess(image)
   return input_tensor.unsqueeze(0) # create a mini-batch as expected by the model
 
 #Runs the CNN on input