mirror of
https://github.com/opencv/opencv_contrib.git
synced 2025-10-19 19:44:14 +08:00
2d69dd1215
Migrate G-API module from main repo to opencv_contrib #3827 Related to https://github.com/opencv/opencv/pull/26469 Required https://github.com/opencv/opencv/pull/26527 CI: https://github.com/opencv/ci-gha-workflow/pull/201 TODO: - [x] Python types generator fix - [x] CI update ### Pull Request Readiness Checklist See details at https://github.com/opencv/opencv/wiki/How_to_contribute#making-a-good-pull-request - [x] I agree to contribute to the project under Apache 2 License. - [x] To the best of my knowledge, the proposed patch is not based on a code under GPL or another license that is incompatible with OpenCV - [ ] The PR is proposed to the proper branch - [ ] There is a reference to the original bug report and related work - [ ] There is accuracy test, performance test and test data in opencv_extra repository, if applicable Patch to opencv_extra has the same branch name. - [ ] The feature is well documented and sample code can be built with the project CMake
61 lines
1.9 KiB
C++
61 lines
1.9 KiB
C++
#include <fstream>
|
|
|
|
#include <opencv2/gapi.hpp>
|
|
#include <opencv2/gapi/core.hpp>
|
|
#include <opencv2/gapi/gframe.hpp>
|
|
|
|
#include <opencv2/gapi/oak/oak.hpp>
|
|
#include <opencv2/gapi/streaming/format.hpp> // BGR accessor
|
|
|
|
#include <opencv2/highgui.hpp> // CommandLineParser
|
|
|
|
const std::string keys =
|
|
"{ h help | | Print this help message }"
|
|
"{ output | output.h265 | Path to the output .h265 video file }";
|
|
|
|
int main(int argc, char *argv[]) {
|
|
cv::CommandLineParser cmd(argc, argv, keys);
|
|
if (cmd.has("help")) {
|
|
cmd.printMessage();
|
|
return 0;
|
|
}
|
|
|
|
const std::string output_name = cmd.get<std::string>("output");
|
|
|
|
cv::gapi::oak::EncoderConfig cfg;
|
|
cfg.profile = cv::gapi::oak::EncoderConfig::Profile::H265_MAIN;
|
|
|
|
cv::GFrame in;
|
|
cv::GArray<uint8_t> encoded = cv::gapi::oak::encode(in, cfg);
|
|
|
|
auto args = cv::compile_args(cv::gapi::oak::ColorCameraParams{}, cv::gapi::oak::kernels());
|
|
|
|
auto pipeline = cv::GComputation(cv::GIn(in), cv::GOut(encoded)).compileStreaming(std::move(args));
|
|
|
|
// Graph execution /////////////////////////////////////////////////////////
|
|
pipeline.setSource(cv::gapi::wip::make_src<cv::gapi::oak::ColorCamera>());
|
|
pipeline.start();
|
|
|
|
std::vector<uint8_t> out_h265_data;
|
|
|
|
std::ofstream out_h265_file;
|
|
out_h265_file.open(output_name, std::ofstream::out | std::ofstream::binary | std::ofstream::trunc);
|
|
|
|
// Pull 300 frames from the camera
|
|
uint32_t frames = 300;
|
|
uint32_t pulled = 0;
|
|
|
|
while (pipeline.pull(cv::gout(out_h265_data))) {
|
|
if (out_h265_file.is_open()) {
|
|
out_h265_file.write(reinterpret_cast<const char*>(out_h265_data.data()),
|
|
out_h265_data.size());
|
|
}
|
|
if (pulled++ == frames) {
|
|
pipeline.stop();
|
|
break;
|
|
}
|
|
}
|
|
|
|
std::cout << "Pipeline finished: " << output_name << " file has been written." << std::endl;
|
|
}
|