$extrastylesheet
Olena  User documentation 2.1
An Image Processing Platform
 All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Groups Pages
sauvola_ms_hdibco.cc
1 // Copyright (C) 2010, 2013 EPITA Research and Development Laboratory
2 // (LRDE)
3 //
4 // This file is part of Olena.
5 //
6 // Olena is free software: you can redistribute it and/or modify it under
7 // the terms of the GNU General Public License as published by the Free
8 // Software Foundation, version 2 of the License.
9 //
10 // Olena is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 // General Public License for more details.
14 //
15 // You should have received a copy of the GNU General Public License
16 // along with Olena. If not, see <http://www.gnu.org/licenses/>.
17 //
18 // As a special exception, you may use this file as part of a free
19 // software project without restriction. Specifically, if other files
20 // instantiate templates or use macros or inline functions from this
21 // file, or you compile this file and link it with other files to produce
22 // an executable, this file does not by itself cause the resulting
23 // executable to be covered by the GNU General Public License. This
24 // exception does not however invalidate any other reasons why the
25 // executable file might be covered by the GNU General Public License.
26 
27 #include <cctype>
28 #include <mln/core/image/image2d.hh>
29 #include <mln/value/int_u8.hh>
30 #include <mln/io/pgm/load.hh>
31 #include <mln/io/ppm/load.hh>
32 #include <mln/io/pbm/save.hh>
33 #include <mln/data/transform.hh>
34 #include <mln/fun/v2v/rgb_to_int_u.hh>
35 
36 #include <scribo/binarization/sauvola_ms.hh>
37 #include <scribo/postprocessing/fill_object_holes.hh>
38 #include <scribo/preprocessing/denoise.hh>
39 
40 int main(int argc, char *argv[])
41 {
42  using namespace mln;
43  using namespace scribo;
44 
45 
46  if (argc != 3)
47  {
48  std::cout << "Usage: " << argv[0] << " <input.{pgm,ppm}> <output.pbm>" << std::endl;
49  std::cout << "--------------" << std::endl;
50  std::cout << "EPITA/LRDE - 2010" << std::endl;
51  return 1;
52  }
53 
54  std::string ext = argv[1];
55  ext = ext.substr(ext.length() - 3, 3);
56  for (unsigned i = 0; i < ext.length(); ++i)
57  ext[i] = tolower(ext[i]);
58 
59  // Load
61 
62  if (ext == "pgm")
63  io::pgm::load(input, argv[1]);
64  else if (ext == "ppm")
65  {
66  image2d<value::rgb8> input_color;
67  io::ppm::load(input_color, argv[1]);
68 
69  // Convert to Gray level image.
70  input = data::transform(input_color, mln::fun::v2v::rgb_to_int_u<8>());
71  }
72  else
73  {
74  std::cout << "Invalid input file format! Expecting PGM or PPM images." << std::endl;
75  return 1;
76  }
77 
78  // Binarize
80  output = scribo::binarization::sauvola_ms(input, 101, 3, 0.4, 0.4, 0.4);
81 
82  output = scribo::preprocessing::denoise_bg(output, c8(), 4);
83  output = scribo::postprocessing::fill_object_holes(output, 0.01f);
84 
85  io::pbm::save(output, argv[2]);
86 }
87 
88