commit f3f6f03bb341b79288f2830a3fceec7c951ff1fe
author Jonas Larsson <ljonas@google.com> Fri Sep 27 16:26:28 2019 -0700
committer Jonas Larsson <ljonas@google.com> Tue Oct 01 13:06:56 2019 -0700
tree cbd290c4fd00d310a10d4fd70d3b3a988e8949ba
parent d409f82d2808f35d39aa0117211cfc9e785337da

Fix PNG image inference

PNG images don't get caps set by GstPbutils.Discoverer so they can't
be used. Instead use the get_width() and get_height() APIs.

Change-Id: I18948c2e1ed9eaf890939933dc90c97cd9eb87e1

edgetpuvision/gstreamer.py

diff --git a/edgetpuvision/gstreamer.py b/edgetpuvision/gstreamer.py
index d553b5c..6e78f22 100644
--- a/edgetpuvision/gstreamer.py
+++ b/edgetpuvision/gstreamer.py
@@ -266,7 +266,7 @@
     filename = os.path.expanduser(source)
     if os.path.isfile(filename):
         info = get_video_info(filename)
-        render_size = caps_size(info.get_caps())
+        render_size = Size(info.get_width(), info.get_height())
         layout = make_layout(inference_size, render_size)
         return layout, file_pipline(info.is_image(), filename, layout, display)