From 139a731d10bf400c904b97cdd0654cc05881321d Mon Sep 17 00:00:00 2001
From: Patrick Simianer <p@simianer.de>
Date: Sat, 1 Apr 2017 21:43:28 +0200
Subject: denoise

---
 denoise.cc | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 denoise.cc

(limited to 'denoise.cc')

diff --git a/denoise.cc b/denoise.cc
new file mode 100644
index 0000000..22f0792
--- /dev/null
+++ b/denoise.cc
@@ -0,0 +1,47 @@
+#include <iostream>
+#include <vector>
+#include "opencv2/opencv.hpp"
+#include "opencv2/photo/photo.hpp"
+#include "cxxopts/src/cxxopts.hpp"
+
+using namespace std;
+using namespace cv;
+
+int
+main (int argc, char** argv)
+{
+  cxxopts::Options opt(argv[0], " - Options");
+  opt.add_options()
+    ("f,file",            "file",                     cxxopts::value<string>())
+    ("b,black_and_white", "image is black and white")
+    ("h,hh",              "h",                        cxxopts::value<float>()->default_value("3.0"))
+    ("C,hColor",          "hColor",                   cxxopts::value<float>()->default_value("3.0"))
+    ("t,template_size",   "template window size",     cxxopts::value<int>()->default_value("7"))
+    ("s,search_size",     "search window size",       cxxopts::value<int>()->default_value("21"))
+    ("T,threads",         "number of threads",        cxxopts::value<int>()->default_value("1"))
+    ("o,output",          "output file",              cxxopts::value<string>());
+
+  opt.parse(argc, argv);
+
+  auto& file = opt["f"].as<string>();
+  auto is_bw = opt.count("b");
+  auto& output = opt["o"].as<string>();
+  float h = opt["h"].as<float>();
+  float hColor = opt["C"].as<float>();
+  float template_size = opt["t"].as<int>();
+  float search_size = opt["s"].as<int>();
+  setNumThreads(opt["T"].as<int>());
+
+  Mat src = imread(file, 1);
+  Mat dst;
+
+  if (is_bw)
+    fastNlMeansDenoising(src, dst, h, template_size, search_size);
+  else
+    fastNlMeansDenoisingColored(src, dst, h, hColor, template_size, search_size);
+
+  imwrite(output, dst);
+
+  return 0;
+}
+
-- 
cgit v1.2.3