Skip to content
Snippets Groups Projects
Commit f71c96f5 authored by Sebastian Hahta's avatar Sebastian Hahta
Browse files

Initialization

parent 267d9e94
No related branches found
No related tags found
1 merge request!15Feature/libsgm mask
Pipeline #11072 failed
...@@ -25,7 +25,7 @@ void FixstarsSGM::compute(const cv::Mat &l, const cv::Mat &r, cv::Mat &disp) { ...@@ -25,7 +25,7 @@ void FixstarsSGM::compute(const cv::Mat &l, const cv::Mat &r, cv::Mat &disp) {
cv::cvtColor(l, lbw, cv::COLOR_BGR2GRAY); cv::cvtColor(l, lbw, cv::COLOR_BGR2GRAY);
cv::cvtColor(r, rbw, cv::COLOR_BGR2GRAY); cv::cvtColor(r, rbw, cv::COLOR_BGR2GRAY);
if (!ssgm_) { if (!ssgm_) { // todo: move to constructor
ssgm_ = new sgm::StereoSGM(l.cols, l.rows, max_disp_, 8, 16, ssgm_ = new sgm::StereoSGM(l.cols, l.rows, max_disp_, 8, 16,
sgm::EXECUTE_INOUT_HOST2HOST, sgm::EXECUTE_INOUT_HOST2HOST,
sgm::StereoSGM::Parameters(10,120,0.95f,true)); sgm::StereoSGM::Parameters(10,120,0.95f,true));
...@@ -60,6 +60,13 @@ void FixstarsSGM::compute(const cv::Mat &l, const cv::Mat &r, cv::Mat &disp) { ...@@ -60,6 +60,13 @@ void FixstarsSGM::compute(const cv::Mat &l, const cv::Mat &r, cv::Mat &disp) {
void FixstarsSGM::setMask(Mat &mask) { void FixstarsSGM::setMask(Mat &mask) {
LOG_ASSERT(mask.type() == CV_8UC1) << "mask type must be CV_8U"; LOG_ASSERT(mask.type() == CV_8UC1) << "mask type must be CV_8U";
if (!ssgm_) { // todo: move to constructor
ssgm_ = new sgm::StereoSGM(mask.cols, mask.rows, max_disp_, 8, 16,
sgm::EXECUTE_INOUT_HOST2HOST,
sgm::StereoSGM::Parameters(10,120,0.95f,true));
}
mask_l_ = mask; mask_l_ = mask;
ssgm_->setMask((uint8_t*) &mask.data, mask.cols); ssgm_->setMask((uint8_t*) mask.data, mask.cols);
} }
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment