Add width and height values to generated SVGs.
Otherwise Firefox cannot render SVG images.
Change-Id: I1e5ded74fb1e3e052fb4dde2300c575988601b74
diff --git a/edgetpuvision/overlays.py b/edgetpuvision/overlays.py
index 3eb9a50..bc7ce51 100644
--- a/edgetpuvision/overlays.py
+++ b/edgetpuvision/overlays.py
@@ -13,7 +13,7 @@
def classification(results, inference_time, inference_rate, size, window):
- x0, y0, _, _ = window
+ x0, y0, w, h = window
lines = [
'Inference time: %.2f ms (%.2f fps)' % (inference_time * 1000, 1.0 / inference_time),
@@ -26,19 +26,19 @@
defs = svg.Defs()
defs += CSS_STYLES
- doc = svg.Svg(viewBox='%s %s %s %s' % window, font_size='26px')
+ doc = svg.Svg(width=w, height=h, viewBox='%s %s %s %s' % window, font_size='26px')
doc += defs
doc += svg.normal_text(lines, x=x0 + 10, y=y0 + 10, font_size_em=1.1)
return str(doc)
def detection(objs, labels, inference_time, inference_rate, size, window):
- x0, y0, _, _ = window
+ x0, y0, w, h = window
defs = svg.Defs()
defs += CSS_STYLES
- doc = svg.Svg(viewBox='%s %s %s %s' % window, font_size='26px')
+ doc = svg.Svg(width=w, height=h, viewBox='%s %s %s %s' % window, font_size='26px')
doc += defs
doc += svg.normal_text((
'Inference time: %.2f ms (%.2f fps)' % (inference_time * 1000, 1.0 / inference_time),
