Enable python-edgetpuvision camera pipeline support
Test: edgetpu_detect --model=/home/mendel/mobilenet_ssd_v2_face_quant_postprocess_edgetpu.tflite
Result at https://drive.google.com/file/d/11f7n5_mDwZXi5R21NcCrmwk8IKDTXyg2
Change-Id: Ie2173103b64357e5a554f3599bb56ae11eff18c1
diff --git a/edgetpuvision/apps.py b/edgetpuvision/apps.py
index bef82e7..2dbcbe6 100644
--- a/edgetpuvision/apps.py
+++ b/edgetpuvision/apps.py
@@ -30,7 +30,7 @@
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--source',
help='/dev/videoN:FMT:WxH:N/D or .mp4 file or image file',
- default='/dev/video0:YUY2:640x480:30/1')
+ default='/dev/video0:RGB:640x480:30/1')
parser.add_argument('--bitrate', type=int, default=1000000,
help='Video streaming bitrate (bit/s)')
parser.add_argument('--loop', default=False, action='store_true',
@@ -56,7 +56,7 @@
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--source',
help='/dev/videoN:FMT:WxH:N/D or .mp4 file or image file',
- default='/dev/video0:YUY2:1280x720:30/1')
+ default='/dev/video0:RGB:1024x768:30/1')
parser.add_argument('--loop', default=False, action='store_true',
help='Loop input video file')
parser.add_argument('--displaymode', type=Display, choices=Display, default=Display.FULLSCREEN,
diff --git a/edgetpuvision/pipelines.py b/edgetpuvision/pipelines.py
index 33daaea..b3d84c7 100644
--- a/edgetpuvision/pipelines.py
+++ b/edgetpuvision/pipelines.py
@@ -29,6 +29,9 @@
Source('v4l2', device=fmt.device),
Caps('video/x-raw', format=fmt.pixel, width=fmt.size.width, height=fmt.size.height,
framerate='%d/%d' % fmt.framerate),
+ Queue(),
+ Filter('videoconvert'),
+ Caps('video/x-raw', format='BGRA'),
]
def display_sink():
@@ -69,7 +72,7 @@
[decoded_file_src(filename),
Tee(name='t')],
[Pad('t'),
- Queue(max_size_buffers=1, leaky='downstream'),
+ Queue(),
display_sink()],
[Pad('t'),
Queue(max_size_buffers=1, leaky='downstream'),
