Create app.py

app.py

@@ -0,0 +1,69 @@
+import numpy as np
+import math
+import matplotlib.pyplot as plt
+import onnxruntime as rt
+import cv2
+import json
+import gradio as gr
+
+# load the labels text file
+labels = json.load(open("labels_map.txt", "r"))
+
+# set image file dimensions to 224x224 by resizing and cropping image from center
+def pre_process_edgetpu(img, dims):
+    output_height, output_width, _ = dims
+    img = resize_with_aspectratio(img, output_height, output_width, inter_pol=cv2.INTER_LINEAR)
+    img = center_crop(img, output_height, output_width)
+    img = np.asarray(img, dtype='float32')
+    # converts jpg pixel value from [0 - 255] to float array [-1.0 - 1.0]
+    img -= [127.0, 127.0, 127.0]
+    img /= [128.0, 128.0, 128.0]
+    return img
+
+# resize the image with a proportional scale
+def resize_with_aspectratio(img, out_height, out_width, scale=87.5, inter_pol=cv2.INTER_LINEAR):
+    height, width, _ = img.shape
+    new_height = int(100. * out_height / scale)
+    new_width = int(100. * out_width / scale)
+    if height > width:
+        w = new_width
+        h = int(new_height * height / width)
+    else:
+        h = new_height
+        w = int(new_width * width / height)
+    img = cv2.resize(img, (w, h), interpolation=inter_pol)
+    return img
+
+# crop the image around the center based on given height and width
+def center_crop(img, out_height, out_width):
+    height, width, _ = img.shape
+    left = int((width - out_width) / 2)
+    right = int((width + out_width) / 2)
+    top = int((height - out_height) / 2)
+    bottom = int((height + out_height) / 2)
+    img = img[top:bottom, left:right]
+    return img
+
+
+
+import onnxruntime as rt
+
+
+sess = rt.InferenceSession('efficientnet-lite4-11.onnx')
+
+def inference(img):
+  img = cv2.imread(img)
+  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+  
+  img = pre_process_edgetpu(img, (224, 224, 3))
+  
+  img_batch = np.expand_dims(img, axis=0)
+
+  results = sess.run(["Softmax:0"], {"images:0": img_batch})[0]
+  result = reversed(results[0].argsort()[-5:])
+  resultdic = {}
+  for r in result:
+      resultdic[labels[str(r)]] = results[0][r]
+  return resultdic
+  
+gr.Interface(inference,gr.inputs.Image(type="filepath"),"label").launch()