tested depth estimation node

uzh-rpg · Oct 29, 2015 · c1b8f78 · c1b8f78
1 parent 4168d27
commit c1b8f78
Show file tree

Hide file tree

Showing 6 changed files with 28 additions and 17 deletions.
diff --git a/include/rmd/depthmap.h b/include/rmd/depthmap.h
@@ -58,6 +58,8 @@ class Depthmap
   void outputDepthmap(cv::Mat &depth_32fc1) const;
   void outputDenoisedDepthmap(cv::Mat &depth_32fc1, float lambda, int iterations);
 
+  static cv::Mat scaleMat(const cv::Mat &depthmap);
+
 private:
   void inputImage(const cv::Mat &img_8uc1);
 

diff --git a/src/depthmap.cpp b/src/depthmap.cpp
@@ -109,3 +109,16 @@ void rmd::Depthmap::outputDenoisedDepthmap(cv::Mat &depth_32fc1, float lambda, i
         lambda,
         iterations);
 }
+
+cv::Mat rmd::Depthmap::scaleMat(const cv::Mat &depthmap)
+{
+  cv::Mat scaled_depthmap = depthmap.clone();
+  double min_val, max_val;
+  cv::minMaxLoc(scaled_depthmap, &min_val, &max_val);
+  cv::Mat converted;
+  scaled_depthmap = (scaled_depthmap - min_val) * 1.0 / (max_val - min_val);
+  scaled_depthmap.convertTo(converted, CV_8UC1, 255);
+  cv::Mat colored(converted.rows, converted.cols, CV_8UC3);
+  cv::cvtColor(converted, colored, CV_GRAY2BGR);
+  return colored;
+}
diff --git a/src/depthmap_node.cpp b/src/depthmap_node.cpp
@@ -60,6 +60,7 @@ void rmd::DepthmapNode::denseInputCallback(
 
   switch (state_) {
   case rmd::State::TAKE_REFERENCE_FRAME:
+  {
     if(depthmap_.setReferenceImage(
          img_8uC1,
          T_world_curr.inv(),
@@ -73,9 +74,18 @@ void rmd::DepthmapNode::denseInputCallback(
       std::cerr << "ERROR: could not set reference image" << std::endl;
     }
     break;
+  }
   case rmd::State::UPDATE:
-      depthmap_.update(img_8uC1, T_world_curr.inv());
+  {
+    depthmap_.update(img_8uC1, T_world_curr.inv());
+#if 1
+    cv::Mat curr_depth;
+    depthmap_.outputDepthmap(curr_depth);
+    cv::imshow("curr_depth", rmd::Depthmap::scaleMat(curr_depth));
+    cv::waitKey(2);
+#endif
     break;
+  }
   default:
     break;
   }

diff --git a/test/dataset.cpp b/test/dataset.cpp
@@ -196,19 +196,6 @@ const rmd::test::DatasetEntry & rmd::test::Dataset::operator()(size_t index) con
   return dataset_.at(index);
 }
 
-cv::Mat rmd::test::Dataset::scaleMat(const cv::Mat &depthmap)
-{
-  cv::Mat scaled_depthmap = depthmap.clone();
-  double min_val, max_val;
-  cv::minMaxLoc(scaled_depthmap, &min_val, &max_val);
-  cv::Mat converted;
-  scaled_depthmap = (scaled_depthmap - min_val) * 1.0 / (max_val - min_val);
-  scaled_depthmap.convertTo(converted, CV_8UC1, 255);
-  cv::Mat colored(converted.rows, converted.cols, CV_8UC3);
-  cv::cvtColor(converted, colored, CV_GRAY2BGR);
-  return colored;
-}
-
 bool rmd::test::Dataset::loadPathFromEnv()
 {
   const char *env_path = std::getenv(data_path_env_var);

diff --git a/test/dataset.h b/test/dataset.h
@@ -72,7 +72,6 @@ class Dataset
   std::vector<DatasetEntry>::const_iterator end() const;
   const DatasetEntry & operator()(size_t index) const;
 
-  static cv::Mat scaleMat(const cv::Mat &depthmap);
   bool loadPathFromEnv();
   static const char * getDataPathEnvVar();
 

diff --git a/test/dataset_main.cpp b/test/dataset_main.cpp
@@ -89,13 +89,13 @@ int main(int argc, char **argv)
   // show depthmap
   cv::Mat result;
   depthmap.outputDepthmap(result);
-  cv::Mat colored = rmd::test::Dataset::scaleMat(result);
+  cv::Mat colored = rmd::Depthmap::scaleMat(result);
   cv::imshow("result", colored);
 
   // denoise
   cv::Mat denoised_result;
   depthmap.outputDenoisedDepthmap(denoised_result, 0.4f, 200);
-  cv::Mat colored_denoised = rmd::test::Dataset::scaleMat(denoised_result);
+  cv::Mat colored_denoised = rmd::Depthmap::scaleMat(denoised_result);
   cv::imshow("denoised_result", colored_denoised);
 
   cv::waitKey();