Skip to content
Snippets Groups Projects
Commit 75def740 authored by Nicolas Pope's avatar Nicolas Pope
Browse files

Iterate over many focal points

parent f8a55170
No related branches found
No related tags found
1 merge request!347Feature buckets experiment
Pipeline #26718 failed
......@@ -38,21 +38,26 @@ void StereoCSF::compute(cv::InputArray l, cv::InputArray r, cv::OutputArray disp
mat2gray(l, impl_->l);
mat2gray(r, impl_->r);
short2 focal_pt = {short(1000), short(800)};
Array2D<float> disp_array(l.cols(), l.rows());
disp_array.toGpuMat().setTo(cv::Scalar(0.0f));
SalientGradient sgl = {focal_pt, 1000, impl_->l.data(), impl_->gl.data(), impl_->temp.data(), impl_->buckets_l.data(), impl_->l.width, impl_->l.height};
parallel1DWarpSM(sgl, l.rows(), l.cols());
SalientGradientGrouped sgr = {impl_->r.data(), impl_->gr.data(), impl_->temp.data(), impl_->buckets_r.data(), impl_->r.width, impl_->r.height};
parallel1DWarpSM(sgr, r.rows(), r.cols());
for (int fx = 200; fx < l.cols()-200; fx += 50) {
for (int fy = 200; fy < l.rows()-200; fy += 50) {
short2 focal_pt = {short(fx), short(fy)};
SalientGradient sgl = {focal_pt, 1000, impl_->l.data(), impl_->gl.data(), impl_->temp.data(), impl_->buckets_l.data(), impl_->l.width, impl_->l.height};
parallel1DWarpSM(sgl, l.rows(), l.cols());
impl_->focal.toGpuMat().setTo(cv::Scalar(0));
FocalCluster fc = {focal_pt, impl_->buckets_l.data(), impl_->buckets_r.data(), impl_->focal.data(), 1024};
parallel1DWarp(fc, l.rows(), 1);
Array2D<float> disp_array(l.cols(), l.rows());
FocalSelector fs = {focal_pt, impl_->buckets_l.data(), impl_->buckets_r.data(), impl_->focal.data(), disp_array.data(), 1024};
parallel1DWarp(fs, l.rows(), 1);
}
}
disp_array.toGpuMat().download(disparity);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment