zakirangwala/detect.py

Created December 1, 2020 06:48

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/zakirangwala/8ab054afa1d8b1855d3bf25dd6c17c20.js"></script>
Save zakirangwala/8ab054afa1d8b1855d3bf25dd6c17c20 to your computer and use it in GitHub Desktop.

Download ZIP

Revisions

zakirangwala created this gist Dec 1, 2020.

detect.py

-Original file line number
+Diff line change
@@ -0,0 +1,6 @@
+    def detect_fn(image):
+        detection_model = load_model()
+        image, shapes = detection_model.preprocess(image)
+        prediction_dict = detection_model.predict(image, shapes)
+        detections = detection_model.postprocess(prediction_dict, shapes)
+        return detections