Done points and depth processor

This commit is contained in:
John Zhao
2018-04-28 14:27:43 +08:00
parent c49342f9f7
commit 2111a183d0
6 changed files with 40 additions and 15 deletions

View File

@@ -17,14 +17,17 @@ std::string DepthProcessor::Name() {
}
Object *DepthProcessor::OnCreateOutput() {
return nullptr;
return new ObjMat();
}
bool DepthProcessor::OnProcess(
Object *const in, Object *const out, Processor *const parent) {
UNUSED(in)
UNUSED(out)
UNUSED(parent)
const ObjMat *input = Object::Cast<ObjMat>(in);
ObjMat *output = Object::Cast<ObjMat>(out);
cv::Mat channels[3 /*input->value.channels()*/];
cv::split(input->value, channels);
channels[2].convertTo(output->value, CV_16UC1);
return true;
}

View File

@@ -1,10 +1,14 @@
#include "api/processor/points_processor.h"
#include <opencv2/calib3d/calib3d.hpp>
#include <glog/logging.h>
#include <utility>
MYNTEYE_BEGIN_NAMESPACE
PointsProcessor::PointsProcessor() : Processor() {
PointsProcessor::PointsProcessor(cv::Mat Q) : Processor(), Q_(std::move(Q)) {
VLOG(2) << __func__;
}
@@ -17,14 +21,15 @@ std::string PointsProcessor::Name() {
}
Object *PointsProcessor::OnCreateOutput() {
return nullptr;
return new ObjMat();
}
bool PointsProcessor::OnProcess(
Object *const in, Object *const out, Processor *const parent) {
UNUSED(in)
UNUSED(out)
UNUSED(parent)
const ObjMat *input = Object::Cast<ObjMat>(in);
ObjMat *output = Object::Cast<ObjMat>(out);
cv::reprojectImageTo3D(input->value, output->value, Q_, true);
return true;
}

View File

@@ -2,6 +2,8 @@
#define MYNTEYE_POINTS_PROCESSOR_H_
#pragma once
#include <opencv2/core/core.hpp>
#include <string>
#include "api/processor/processor.h"
@@ -12,7 +14,7 @@ class PointsProcessor : public Processor {
public:
static constexpr auto &&NAME = "PointsProcessor";
PointsProcessor();
explicit PointsProcessor(cv::Mat Q);
virtual ~PointsProcessor();
std::string Name() override;
@@ -21,6 +23,9 @@ class PointsProcessor : public Processor {
Object *OnCreateOutput() override;
bool OnProcess(
Object *const in, Object *const out, Processor *const parent) override;
private:
cv::Mat Q_;
};
MYNTEYE_END_NAMESPACE

View File

@@ -23,6 +23,9 @@ class RectifyProcessor : public Processor {
std::string Name() override;
cv::Mat R1, P1, R2, P2, Q;
cv::Mat map11, map12, map21, map22;
protected:
Object *OnCreateOutput() override;
bool OnProcess(
@@ -31,9 +34,6 @@ class RectifyProcessor : public Processor {
private:
void InitParams(
Intrinsics in_left, Intrinsics in_right, Extrinsics ex_left_to_right);
cv::Mat R1, P1, R2, P2, Q;
cv::Mat map11, map12, map21, map22;
};
MYNTEYE_END_NAMESPACE

View File

@@ -367,7 +367,8 @@ void Synthetic::InitProcessors() {
auto &&disparity_processor = std::make_shared<DisparityProcessor>();
auto &&disparitynormalized_processor =
std::make_shared<DisparityNormalizedProcessor>();
auto &&points_processor = std::make_shared<PointsProcessor>();
auto &&points_processor =
std::make_shared<PointsProcessor>(rectify_processor->Q);
auto &&depth_processor = std::make_shared<DepthProcessor>();
using namespace std::placeholders; // NOLINT