diff --git a/demos/face_recognition_demo/python/face_recognition_demo.py b/demos/face_recognition_demo/python/face_recognition_demo.py index 867014d8fb7..7347c57c273 100755 --- a/demos/face_recognition_demo/python/face_recognition_demo.py +++ b/demos/face_recognition_demo/python/face_recognition_demo.py @@ -207,6 +207,10 @@ def center_crop(frame, crop_size): def main(): args = build_argparser().parse_args() + for arg_name, device in [('-d_fd', args.d_fd), ('-d_lm', args.d_lm), ('-d_reid', args.d_reid)]: + if device == 'HETERO': + raise ValueError('{} requires sub-devices, e.g. HETERO:GPU,CPU'.format(arg_name)) + cap = open_images_capture(args.input, args.loop) frame_processor = FrameProcessor(args)