提交 bc0d4585 编写于 作者: A Alexander Alekhin

Merge pull request #21984 from xiong-jie-y:add_make_capture_for_camera_stream

......@@ -119,6 +119,12 @@ GAPI_EXPORTS_W cv::Ptr<IStreamSource> inline make_capture_src(const std::string&
return make_src<GCaptureSource>(path);
}
// NB: Overload for using from python
GAPI_EXPORTS_W cv::Ptr<IStreamSource> inline make_capture_src(const int id)
{
return make_src<GCaptureSource>(id);
}
} // namespace wip
} // namespace gapi
} // namespace cv
......
......@@ -27,7 +27,7 @@ def build_argparser():
parser = argparse.ArgumentParser(description='This is an OpenCV-based version of Gaze Estimation example')
parser.add_argument('--input',
help='Path to the input video file')
help='Path to the input video file or camera device number')
parser.add_argument('--out',
help='Path to the output video file')
parser.add_argument('--facem',
......@@ -323,7 +323,11 @@ if __name__ == '__main__':
# ------------------------Execution part------------------------
ccomp = comp.compileStreaming(args=cv.gapi.compile_args(kernels, nets))
source = cv.gapi.wip.make_capture_src(ARGUMENTS.input)
if ARGUMENTS.input.isdigit():
source = cv.gapi.wip.make_capture_src(int(ARGUMENTS.input))
else:
source = cv.gapi.wip.make_capture_src(ARGUMENTS.input)
ccomp.setSource(cv.gin(source))
ccomp.start()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册