From ec365de429f2dbbd84488e55115c86bcafc380b9 Mon Sep 17 00:00:00 2001
From: minoring <minho1350@gmail.com>
Date: Mon, 3 May 2021 23:07:17 +0900
Subject: [PATCH] =?UTF-8?q?=EA=B5=AC=ED=98=84:=20=EB=9D=BC=EC=A6=88?=
 =?UTF-8?q?=EB=B2=A0=EB=A6=AC=ED=8C=8C=EC=9D=B4=20tflite=20mobilenet=20inf?=
 =?UTF-8?q?erence=20=EA=B5=AC=ED=98=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 deep_learning/pi/install.sh       | 15 +++++++++
 deep_learning/pi/run_inference.py | 52 +++++++++++++++++++++++++++++++
 2 files changed, 67 insertions(+)
 create mode 100644 deep_learning/pi/install.sh
 create mode 100644 deep_learning/pi/run_inference.py

diff --git a/deep_learning/pi/install.sh b/deep_learning/pi/install.sh
new file mode 100644
index 00000000..b857b9c7
--- /dev/null
+++ b/deep_learning/pi/install.sh
@@ -0,0 +1,15 @@
+#!/bin/bash
+
+# TFLite runtime
+echo "deb https://packages.cloud.google.com/apt coral-edgetpu-stable main" | sudo tee /etc/apt/sources.list.d/coral-edgetpu.list
+curl https://packages.cloud.google.com/apt/doc/apt-key.gpg | sudo apt-key add -
+sudo apt-get update
+sudo apt-get install python3-tflite-runtime
+
+# OpenCV
+sudo apt-get -y install libjpeg-dev libtiff5-dev libjasper-dev libpng12-dev
+sudo apt-get -y install libavcodec-dev libavformat-dev libswscale-dev libv4l-dev
+sudo apt-get -y install libxvidcore-dev libx264-dev
+sudo apt-get -y install qt4-dev-tools libatlas-base-dev
+sudo apt-get -y install libhdf5-dev libqtgui4  libqt4-test
+pip3 install opencv-python
diff --git a/deep_learning/pi/run_inference.py b/deep_learning/pi/run_inference.py
new file mode 100644
index 00000000..cc368ff5
--- /dev/null
+++ b/deep_learning/pi/run_inference.py
@@ -0,0 +1,52 @@
+import argparse
+import os
+import time
+from pathlib import Path
+
+import tflite_runtime.interpreter as tflite
+import cv2
+import numpy as np
+
+
+def read_input_images(data_path):
+  img_files = data_path.rglob('*.jpg')
+  imgs = [cv2.imread(str(img_file)) for img_file in img_files]
+  imgs = [cv2.cvtColor(img, cv2.COLOR_BGR2RGB) for img in imgs]
+
+  return imgs
+
+
+def main():
+  parser = argparse.ArgumentParser()
+  parser.add_argument('-m', '--model_file', help="path to tflite model file")
+  parser.add_argument('-d',
+                      '--data_path',
+                      help="path to dataset to run inference")
+  args = parser.parse_args()
+
+  interpreter = tflite.Interpreter(model_path=args.model_file)
+  interpreter.allocate_tensors()
+
+  input_details = interpreter.get_input_details()
+  output_details = interpreter.get_output_details()
+  input_shape = input_details[0]['shape']
+
+  images = read_input_images(Path(args.data_path))
+
+  for image in images:
+    input_data = np.expand_dims(image, axis=0)
+    input_data = np.float32(input_data)
+
+    interpreter.set_tensor(input_details[0]['index'], input_data)
+    start_time = time.time()
+    interpreter.invoke()
+    stop_time = time.time()
+
+    print('Inference time: {:.3f}'.format((stop_time - start_time) * 1000))
+
+    prediction = interpreter.get_tensor(output_details[0]['index'])
+    print(f'Prediction: {prediction[0]}')
+
+
+if __name__ == "__main__":
+  main()