Commit 3abd9417 authored by Alexey Spizhevoy's avatar Alexey Spizhevoy

Minor changes in videostab sample

parent bde74562
...@@ -147,21 +147,21 @@ int main(int argc, const char **argv) ...@@ -147,21 +147,21 @@ int main(int argc, const char **argv)
if (!cmd.get<string>("outlier-ratio").empty()) if (!cmd.get<string>("outlier-ratio").empty())
{ {
RansacParams ransacParams = motionEstimator->ransacParams(); RansacParams ransacParams = motionEstimator->ransacParams();
ransacParams.eps = atof(cmd.get<string>("outlier-ratio").c_str()); ransacParams.eps = cmd.get<float>("outlier-ratio");
motionEstimator->setRansacParams(ransacParams); motionEstimator->setRansacParams(ransacParams);
} }
if (!cmd.get<string>("min-inlier-ratio").empty()) if (!cmd.get<string>("min-inlier-ratio").empty())
motionEstimator->setMinInlierRatio(atof(cmd.get<string>("min-inlier-ratio").c_str())); motionEstimator->setMinInlierRatio(cmd.get<float>("min-inlier-ratio"));
stabilizer->setMotionEstimator(motionEstimator); stabilizer->setMotionEstimator(motionEstimator);
int smoothRadius = -1; int smoothRadius = -1;
float smoothStdev = -1; float smoothStdev = -1;
if (!cmd.get<string>("radius").empty()) if (!cmd.get<string>("radius").empty())
smoothRadius = atoi(cmd.get<string>("radius").c_str()); smoothRadius = cmd.get<int>("radius");
if (!cmd.get<string>("stdev").empty()) if (!cmd.get<string>("stdev").empty())
smoothStdev = atof(cmd.get<string>("stdev").c_str()); smoothStdev = cmd.get<float>("stdev");
if (smoothRadius > 0 && smoothStdev > 0) if (smoothRadius > 0 && smoothStdev > 0)
stabilizer->setMotionFilter(new GaussianMotionFilter(smoothRadius, smoothStdev)); stabilizer->setMotionFilter(new GaussianMotionFilter(smoothRadius, smoothStdev));
else if (smoothRadius > 0 && smoothStdev < 0) else if (smoothRadius > 0 && smoothStdev < 0)
...@@ -171,7 +171,7 @@ int main(int argc, const char **argv) ...@@ -171,7 +171,7 @@ int main(int argc, const char **argv)
{ {
WeightingDeblurer *deblurer = new WeightingDeblurer(); WeightingDeblurer *deblurer = new WeightingDeblurer();
if (!cmd.get<string>("deblur-sens").empty()) if (!cmd.get<string>("deblur-sens").empty())
deblurer->setSensitivity(atof(cmd.get<string>("deblur-sens").c_str())); deblurer->setSensitivity(cmd.get<float>("deblur-sens"));
stabilizer->setDeblurer(deblurer); stabilizer->setDeblurer(deblurer);
} }
...@@ -179,7 +179,7 @@ int main(int argc, const char **argv) ...@@ -179,7 +179,7 @@ int main(int argc, const char **argv)
stabilizer->setEstimateTrimRatio(cmd.get<string>("est-trim") == "yes"); stabilizer->setEstimateTrimRatio(cmd.get<string>("est-trim") == "yes");
if (!cmd.get<string>("trim-ratio").empty()) if (!cmd.get<string>("trim-ratio").empty())
stabilizer->setTrimRatio(atof(cmd.get<string>("trim-ratio").c_str())); stabilizer->setTrimRatio(cmd.get<float>("trim-ratio"));
stabilizer->setInclusionConstraint(cmd.get<string>("incl-constr") == "yes"); stabilizer->setInclusionConstraint(cmd.get<string>("incl-constr") == "yes");
...@@ -195,7 +195,7 @@ int main(int argc, const char **argv) ...@@ -195,7 +195,7 @@ int main(int argc, const char **argv)
{ {
ConsistentMosaicInpainter *inpainter = new ConsistentMosaicInpainter(); ConsistentMosaicInpainter *inpainter = new ConsistentMosaicInpainter();
if (!cmd.get<string>("mosaic-stdev").empty()) if (!cmd.get<string>("mosaic-stdev").empty())
inpainter->setStdevThresh(atof(cmd.get<string>("mosaic-stdev").c_str())); inpainter->setStdevThresh(cmd.get<float>("mosaic-stdev"));
inpainters->pushBack(inpainter); inpainters->pushBack(inpainter);
} }
if (cmd.get<string>("motion-inpaint") == "yes") if (cmd.get<string>("motion-inpaint") == "yes")
...@@ -210,7 +210,7 @@ int main(int argc, const char **argv) ...@@ -210,7 +210,7 @@ int main(int argc, const char **argv)
outputPath = cmd.get<string>("output"); outputPath = cmd.get<string>("output");
if (!cmd.get<string>("fps").empty()) if (!cmd.get<string>("fps").empty())
outputFps = atoi(cmd.get<string>("fps").c_str()); outputFps = cmd.get<double>("fps");
// run video processing // run video processing
run(); run();
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment