mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-24 03:00:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add option to force reopen camera in interactive calibration tool

Browse Source
This commit is contained in:
Smirnov Egor 2022-07-04 15:56:28 +03:00
parent 3bf22d0024
commit 08696d92ea
4 changed files with 50 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
apps/interactive-calibration/calibCommon.hpp
View File

@ -95,6 +95,7 @@ namespace calib
int minFramesNum;
bool saveFrames;
float zoom;
bool forceReopen;
captureParameters()
{

69
apps/interactive-calibration/calibPipeline.cpp
View File

@ -6,6 +6,7 @@
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/core/utils/logger.hpp>
#include <stdexcept>
@ -30,37 +31,61 @@ CalibPipeline::CalibPipeline(captureParameters params) :
PipelineExitStatus CalibPipeline::start(std::vector<cv::Ptr<FrameProcessor> > processors)
{
if(mCaptureParams.source == Camera && !mCapture.isOpened())
{
mCapture.open(mCaptureParams.camID);
cv::Size maxRes = getCameraResolution();
cv::Size neededRes = mCaptureParams.cameraResolution;
auto open_camera = [this] () {
if(mCaptureParams.source == Camera)
{
mCapture.open(mCaptureParams.camID);
cv::Size maxRes = getCameraResolution();
cv::Size neededRes = mCaptureParams.cameraResolution;
if(maxRes.width < neededRes.width) {
double aR = (double)maxRes.width / maxRes.height;
mCapture.set(cv::CAP_PROP_FRAME_WIDTH, neededRes.width);
mCapture.set(cv::CAP_PROP_FRAME_HEIGHT, neededRes.width/aR);
if(maxRes.width < neededRes.width) {
double aR = (double)maxRes.width / maxRes.height;
mCapture.set(cv::CAP_PROP_FRAME_WIDTH, neededRes.width);
mCapture.set(cv::CAP_PROP_FRAME_HEIGHT, neededRes.width/aR);
}
else if(maxRes.height < neededRes.height) {
double aR = (double)maxRes.width / maxRes.height;
mCapture.set(cv::CAP_PROP_FRAME_HEIGHT, neededRes.height);
mCapture.set(cv::CAP_PROP_FRAME_WIDTH, neededRes.height*aR);
}
else {
mCapture.set(cv::CAP_PROP_FRAME_HEIGHT, neededRes.height);
mCapture.set(cv::CAP_PROP_FRAME_WIDTH, neededRes.width);
}
mCapture.set(cv::CAP_PROP_AUTOFOCUS, 0);
}
else if(maxRes.height < neededRes.height) {
double aR = (double)maxRes.width / maxRes.height;
mCapture.set(cv::CAP_PROP_FRAME_HEIGHT, neededRes.height);
mCapture.set(cv::CAP_PROP_FRAME_WIDTH, neededRes.height*aR);
}
else {
mCapture.set(cv::CAP_PROP_FRAME_HEIGHT, neededRes.height);
mCapture.set(cv::CAP_PROP_FRAME_WIDTH, neededRes.width);
}
mCapture.set(cv::CAP_PROP_AUTOFOCUS, 0);
else if (mCaptureParams.source == File)
mCapture.open(mCaptureParams.videoFileName);
};
if(!mCapture.isOpened()) {
open_camera();
}
else if (mCaptureParams.source == File && !mCapture.isOpened())
mCapture.open(mCaptureParams.videoFileName);
mImageSize = cv::Size((int)mCapture.get(cv::CAP_PROP_FRAME_WIDTH), (int)mCapture.get(cv::CAP_PROP_FRAME_HEIGHT));
if(!mCapture.isOpened())
throw std::runtime_error("Unable to open video source");
cv::Mat frame, processedFrame, resizedFrame;
while(mCapture.grab()) {
while (true) {
if (!mCapture.grab())
{
if (!mCaptureParams.forceReopen)
{
CV_LOG_ERROR(NULL, "VideoCapture error: could not grab the frame.");
break;
}
CV_LOG_INFO(NULL, "VideoCapture error: trying to reopen...");
do
{
open_camera();
} while (!mCapture.isOpened() || !mCapture.grab());
CV_LOG_INFO(NULL, "VideoCapture error: reopened successfully.");
auto newSize = cv::Size((int)mCapture.get(cv::CAP_PROP_FRAME_WIDTH), (int)mCapture.get(cv::CAP_PROP_FRAME_HEIGHT));
CV_CheckEQ(mImageSize, newSize, "Camera image size changed after reopening.");
}
mCapture.retrieve(frame);
if(mCaptureParams.flipVertical)
cv::flip(frame, frame, -1);

1
apps/interactive-calibration/main.cpp
View File

@ -42,6 +42,7 @@ const std::string keys =
"{pf | defaultConfig.xml| Advanced application parameters}"
"{save_frames | false | Save frames that contribute to final calibration}"
"{zoom | 1 | Zoom factor applied to the image}"
"{force_reopen | false | Forcefully reopen camera in case of errors}"
"{help | | Print help}";
bool calib::showOverlayMessage(const std::string& message)

1
apps/interactive-calibration/parametersController.cpp
View File

@ -91,6 +91,7 @@ bool calib::parametersController::loadFromParser(cv::CommandLineParser &parser)
mCapParams.templDst = parser.get<float>("dst");
mCapParams.saveFrames = parser.has("save_frames");
mCapParams.zoom = parser.get<float>("zoom");
mCapParams.forceReopen = parser.get<bool>("force_reopen");
if(!checkAssertion(mCapParams.squareSize > 0, "Distance between corners or circles must be positive"))
return false;