From 6f033ea30a0f2edf9a7f059d3678e68ed52b4dd9 Mon Sep 17 00:00:00 2001
From: dustcollector12 <44067540+dustcollector12@users.noreply.github.com>
Date: Fri, 13 Nov 2020 16:00:12 +0100
Subject: [PATCH] enable local images for efficientnet.py (#116)

---
 examples/efficientnet.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/examples/efficientnet.py b/examples/efficientnet.py
index 4a24f83fc5..6ab380ad11 100644
--- a/examples/efficientnet.py
+++ b/examples/efficientnet.py
@@ -234,7 +234,10 @@ if __name__ == "__main__":
     cap.release()
     cv2.destroyAllWindows()
   else:
-    img = Image.open(io.BytesIO(fetch(url)))
+    if url.startswith('http'):
+      img = Image.open(io.BytesIO(fetch(url)))
+    else:
+      img = Image.open(url)
     st = time.time()
     out, _ = infer(model, img)
     print(np.argmax(out.data), np.max(out.data), lbls[np.argmax(out.data)])