Add --bitrate command line param.
Change-Id: I71457343c5745b538de69fdc048eb51cad6f126a
diff --git a/edgetpuvision/server.py b/edgetpuvision/server.py
index e1e35da..bb3c0aa 100644
--- a/edgetpuvision/server.py
+++ b/edgetpuvision/server.py
@@ -9,6 +9,9 @@
logging.basicConfig(level=logging.INFO)
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+ parser.add_argument('--bitrate', type=int, default=1000000,
+ help='Video streaming bitrate (bit/s)')
+
parser.add_argument('--source',
help='/dev/videoN:FMT:WxH:N/D or .mp4 file or image file',
default='/dev/video0:YUY2:1280x720:30/1')
@@ -19,7 +22,7 @@
camera = make_camera(args.source, next(gen))
assert camera is not None
- with StreamingServer(camera) as server:
+ with StreamingServer(camera, args.bitrate) as server:
def on_image(tensor, layout):
overlay = gen.send((tensor, layout, None))
server.send_overlay(overlay)
