mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-01-18 22:44:02 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add make_capture_src for video stream to Python

Browse Source
This commit is contained in:
xiong-jie-y 2022-05-15 13:00:01 +09:00
parent 8d0fbc6a1e
commit c929f1b62f
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/gapi/include/opencv2/gapi/streaming/cap.hpp
View File

@ -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

8
modules/gapi/misc/python/samples/gaze_estimation.py
View File

@ -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()