Update utils.py

utils.py

@@ -7,17 +7,18 @@ model = core.read_model(model='models/horizontal-text-detection-0001.xml')
 compiled_model = core.compile_model(model = model, device_name="CPU")
 input_layer = compiled_model.input(0)
 output_layer = compiled_model.output(0)
-#output_layer = compiled_model.output("boxes")
 
-def preprocess(image, input_layer):
+def preprocess_data(image, input_layer):
     N, C, H, W = input_layer.shape
     resized_image = cv2.resize(image, (W, H))
     input_image = np.expand_dims(resized_image.transpose(2, 0, 1), 0)
     return input_image, resized_image
 
 def predict_image(image, conf_threshold):
-    input_image, resized_image = preprocess(image, input_layer)
-    boxes = compiled_model([input_image])[output_layer]
+    input_image, resized_image = preprocess_data(image, input_layer)
+    output_key = compiled_model.output("boxes")
+    boxes = compiled_model([input_image])[output_key]
+    # Remove zero only boxes.
     boxes = boxes[~np.all(boxes == 0, axis=1)]
     return boxes, resized_image