* src/preprocessing/Makefile.am, * src/text/Makefile.am, * src/Makefile.am: Update. * src/multi_scale/Makefile.am, * src/multi_scale/find_lines.cc * src/text/pbm_recognition.cc * src/text/recognition.cc * src/text_in_article_preprocess.cc * src/text_in_doc.cc * src/text_in_doc_ppm.cc * src/text_in_photo.cc * src/text_in_photo_invert.cc * src/text_in_photo_pbm_fast.cc * src/text_in_photo_ppm.cc * test.cc: Remove. Deprecated. * src/text_in_article_pbm.cc: Rename as... * src/pbm_text_in_doc.cc: ... this. Make use of the new toolchain routines. * src/text/pbm_lines_recognition.cc, * src/text_in_doc_preprocess.cc: New. * src/text_in_photo_fast.cc: Rename as... * src/text_in_picture.cc: ... this. * src/preprocessing/subsample.cc: Make use of io::magick. --- scribo/ChangeLog | 33 ++ scribo/src/Makefile.am | 78 +-- scribo/src/multi_scale/Makefile.am | 26 - scribo/src/multi_scale/find_lines.cc | 180 ------- scribo/src/pbm_text_in_doc.cc | 132 +++++ scribo/src/preprocessing/Makefile.am | 26 +- scribo/src/preprocessing/subsample.cc | 16 +- scribo/src/text/Makefile.am | 18 +- .../{recognition.cc => pbm_lines_recognition.cc} | 92 ++-- scribo/src/text/pbm_recognition.cc | 20 +- scribo/src/text_in_article_pbm.cc | 381 -------------- scribo/src/text_in_doc.cc | 304 ----------- scribo/src/text_in_doc_ppm.cc | 322 ------------ ...cle_preprocess.cc => text_in_doc_preprocess.cc} | 18 +- scribo/src/text_in_photo.cc | 250 --------- scribo/src/text_in_photo_invert.cc | 245 --------- scribo/src/text_in_photo_pbm_fast.cc | 441 ---------------- scribo/src/text_in_photo_ppm.cc | 532 -------------------- .../{text_in_photo_fast.cc => text_in_picture.cc} | 2 + scribo/test.cc | 44 -- 20 files changed, 312 insertions(+), 2848 deletions(-) delete mode 100644 scribo/src/multi_scale/Makefile.am delete mode 100644 scribo/src/multi_scale/find_lines.cc create mode 100644 scribo/src/pbm_text_in_doc.cc rename scribo/src/text/{recognition.cc => pbm_lines_recognition.cc} (52%) delete mode 100644 scribo/src/text_in_article_pbm.cc delete mode 100644 scribo/src/text_in_doc.cc delete mode 100644 scribo/src/text_in_doc_ppm.cc rename scribo/src/{text_in_article_preprocess.cc => text_in_doc_preprocess.cc} (84%) delete mode 100644 scribo/src/text_in_photo.cc delete mode 100644 scribo/src/text_in_photo_invert.cc delete mode 100644 scribo/src/text_in_photo_pbm_fast.cc delete mode 100644 scribo/src/text_in_photo_ppm.cc rename scribo/src/{text_in_photo_fast.cc => text_in_picture.cc} (99%) delete mode 100644 scribo/test.cc diff --git a/scribo/ChangeLog b/scribo/ChangeLog index f091afa..de7acfa 100644 --- a/scribo/ChangeLog +++ b/scribo/ChangeLog @@ -1,5 +1,38 @@ 2010-05-25 Guillaume Lazzara &lt;z(a)lrde.epita.fr&gt; + Cleanup sample tools. + + * src/preprocessing/Makefile.am, + * src/text/Makefile.am, + * src/Makefile.am: Update. + + * src/multi_scale/Makefile.am, + * src/multi_scale/find_lines.cc + * src/text/pbm_recognition.cc + * src/text/recognition.cc + * src/text_in_article_preprocess.cc + * src/text_in_doc.cc + * src/text_in_doc_ppm.cc + * src/text_in_photo.cc + * src/text_in_photo_invert.cc + * src/text_in_photo_pbm_fast.cc + * src/text_in_photo_ppm.cc + * test.cc: Remove. Deprecated. + + * src/text_in_article_pbm.cc: Rename as... + * src/pbm_text_in_doc.cc: ... this. Make use of the new toolchain + routines. + + * src/text/pbm_lines_recognition.cc, + * src/text_in_doc_preprocess.cc: New. + + * src/text_in_photo_fast.cc: Rename as... + * src/text_in_picture.cc: ... this. + + * src/preprocessing/subsample.cc: Make use of io::magick. + +2010-05-25 Guillaume Lazzara &lt;z(a)lrde.epita.fr&gt; + * toolchain/text_in_doc.hh: Introduce a new routine for extracting text in a document. diff --git a/scribo/src/Makefile.am b/scribo/src/Makefile.am index 08b4bd9..e910f53 100644 --- a/scribo/src/Makefile.am +++ b/scribo/src/Makefile.am @@ -25,81 +25,51 @@ SUBDIRS = \ primitive \ filter \ misc \ - multi_scale \ + nuxeo_xwiki \ preprocessing \ table \ text bin_PROGRAMS = \ - pbm_lines_in_doc \ - text_in_photo \ - text_in_photo_ppm \ - text_in_photo_pbm_fast \ - text_in_photo_invert + pbm_lines_in_doc pbm_lines_in_doc_SOURCES = pbm_lines_in_doc.cc -text_in_photo_SOURCES = text_in_photo.cc -text_in_photo_ppm_SOURCES = text_in_photo_ppm.cc -text_in_photo_pbm_fast_SOURCES = text_in_photo_pbm_fast.cc -text_in_photo_invert_SOURCES = text_in_photo_invert.cc - if HAVE_TESSERACT -if HAVE_TIFF - - bin_PROGRAMS += text_in_doc - text_in_doc_SOURCES = text_in_doc.cc - text_in_doc_CPPFLAGS = $(AM_CPPFLAGS) \ - $(TESSERACT_CPPFLAGS) \ - $(TIFF_CPPFLAGS) - text_in_doc_LDFLAGS = $(AM_LDFLAGS) \ - $(TESSERACT_LDFLAGS) \ - $(TIFF_LDFLAGS) \ - -lpthread -lhpdf - - bin_PROGRAMS += text_in_doc_ppm - text_in_doc_ppm_SOURCES = text_in_doc_ppm.cc - text_in_doc_ppm_CPPFLAGS = $(AM_CPPFLAGS) \ - $(TESSERACT_CPPFLAGS) \ - $(TIFF_CPPFLAGS) - text_in_doc_ppm_LDFLAGS = $(AM_LDFLAGS) \ - $(TESSERACT_LDFLAGS) \ - $(TIFF_LDFLAGS) \ - -lpthread -lhpdf - - - bin_PROGRAMS += text_in_article_pbm - text_in_article_pbm_CPPFLAGS = $(AM_CPPFLAGS) \ + + bin_PROGRAMS += pbm_text_in_doc + pbm_text_in_doc_CPPFLAGS = $(AM_CPPFLAGS) -g -ggdb \ $(TESSERACT_CPPFLAGS) \ $(TIFF_CPPFLAGS) - text_in_article_pbm_LDFLAGS = $(AM_LDFLAGS) \ + pbm_text_in_doc_LDFLAGS = $(AM_LDFLAGS) -g -ggdb\ $(TESSERACT_LDFLAGS) \ $(TIFF_LDFLAGS) \ -lpthread # -lhpdf - text_in_article_pbm_SOURCES = text_in_article_pbm.cc + pbm_text_in_doc_SOURCES = pbm_text_in_doc.cc + +endif HAVE_TESSERACT - bin_PROGRAMS += text_in_article_preprocess - text_in_article_preprocess_CPPFLAGS = $(AM_CPPFLAGS) \ - `Magick++-config --cppflags` - text_in_article_preprocess_LDFLAGS = $(AM_LDFLAGS) \ - -lpthread `Magick++-config --libs` - text_in_article_preprocess_SOURCES = text_in_article_preprocess.cc +if HAVE_MAGICKXX + bin_PROGRAMS += text_in_doc_preprocess + text_in_doc_preprocess_CPPFLAGS = $(AM_CPPFLAGS) \ + `Magick++-config --cppflags` + text_in_doc_preprocess_LDFLAGS = $(AM_LDFLAGS) \ + -lpthread `Magick++-config --libs` + text_in_doc_preprocess_SOURCES = text_in_doc_preprocess.cc - bin_PROGRAMS += text_in_photo_fast - text_in_photo_fast_SOURCES = text_in_photo_fast.cc - text_in_photo_fast_CPPFLAGS = $(AM_CPPFLAGS) \ - $(TESSERACT_CPPFLAGS) \ - `Magick++-config --cppflags` - text_in_photo_fast_LDFLAGS = $(AM_LDFLAGS) \ - $(TESSERACT_LDFLAGS) \ - -lpthread `Magick++-config --libs` + bin_PROGRAMS += text_in_picture + text_in_picture_SOURCES = text_in_picture.cc + text_in_picture_CPPFLAGS = $(AM_CPPFLAGS) \ + `Magick++-config --cppflags` -endif HAVE_TIFF -endif HAVE_TESSERACT + text_in_picture_LDFLAGS = $(AM_LDFLAGS) \ + -lpthread `Magick++-config --libs` + +endif HAVE_MAGICKXX diff --git a/scribo/src/multi_scale/Makefile.am b/scribo/src/multi_scale/Makefile.am deleted file mode 100644 index f3dfb42..0000000 --- a/scribo/src/multi_scale/Makefile.am +++ /dev/null @@ -1,26 +0,0 @@ -# Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE). -# -# This file is part of Olena. -# -# Olena is free software: you can redistribute it and/or modify it under -# the terms of the GNU General Public License as published by the Free -# Software Foundation, version 2 of the License. -# -# Olena is distributed in the hope that it will be useful, -# but WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU -# General Public License for more details. -# -# You should have received a copy of the GNU General Public License -# along with Olena. If not, see <http://www.gnu.org/licenses/>. -# - -## Process this file through Automake to create Makefile.in. - -include $(top_srcdir)/scribo/scribo.mk - -bin_PROGRAMS = \ - find_lines - -find_lines_SOURCES = find_lines.cc - diff --git a/scribo/src/multi_scale/find_lines.cc b/scribo/src/multi_scale/find_lines.cc deleted file mode 100644 index c5a2f97..0000000 --- a/scribo/src/multi_scale/find_lines.cc +++ /dev/null @@ -1,180 +0,0 @@ -// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE) -// -// This file is part of Olena. -// -// Olena is free software: you can redistribute it and/or modify it under -// the terms of the GNU General Public License as published by the Free -// Software Foundation, version 2 of the License. -// -// Olena is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU -// General Public License for more details. -// -// You should have received a copy of the GNU General Public License -// along with Olena. If not, see <http://www.gnu.org/licenses/>. -// -// As a special exception, you may use this file as part of a free -// software project without restriction. Specifically, if other files -// instantiate templates or use macros or inline functions from this -// file, or you compile this file and link it with other files to produce -// an executable, this file does not by itself cause the resulting -// executable to be covered by the GNU General Public License. This -// exception does not however invalidate any other reasons why the -// executable file might be covered by the GNU General Public License. - -#include <mln/core/alias/neighb2d.hh> -#include <mln/core/image/image2d.hh> -#include <mln/data/convert.hh> -#include <mln/debug/superpose.hh> -#include <mln/io/pbm/all.hh> -#include <mln/io/ppm/save.hh> -#include <mln/pw/all.hh> -#include <mln/subsampling/subsampling.hh> -#include <mln/value/label_16.hh> -#include <mln/value/rgb8.hh> - -#include <mln/world/binary_2d/enlarge.hh> - -#include <scribo/debug/usage.hh> - -#include <scribo/core/object_image.hh> -#include <scribo/primitive/extract/lines_h_pattern.hh> -#include <scribo/primitive/extract/lines_v_pattern.hh> - -#include <scribo/filter/objects_h_thin.hh> -#include <scribo/filter/objects_v_thin.hh> - -#include <sandbox/theo/Rd/sequential.hh> - -#include <mln/morpho/erosion.hh> - -const char *args_desc[][2] = -{ - { "input.pbm", "A binary image." }, - { "length", " Minimum line length." }, - {0, 0} -}; - - -namespace mln -{ - - template <typename I> - mln_concrete(I) - process(const I& input, const std::string& filename, - unsigned length, unsigned delta, unsigned ratio) - { - (void) filename; - - I hlines = scribo::primitive::extract::lines_h_pattern(input, - length, - delta); - - value::label_16 nhlines; - hlines = scribo::filter::objects_v_thin(hlines, c8(), - nhlines, delta * ratio); -// I vlines = scribo::primitive::extract::lines_v_pattern(input, -// length, -// delta); - - -// value::label_16 nvlines; -// vlines = scribo::filter::objects_h_thin(vlines, c8(), -// nvlines, delta * ratio); - -// image2d<value::rgb8> out = debug::superpose(input, hlines, literal::red); -// out = debug::superpose(out, vlines, literal::green); -// io::ppm::save(out, filename); - - return hlines; - } - - - template <typename I> - mln_concrete(I) - merge_results(const I& input, - const I& out, const I& out_sub2x, const I& out_sub4x) - { - mln_concrete(I) output; - initialize(output, input); - - - - return output; - } - - -} // end of namespace mln - - - -int main(int argc, char *argv[]) -{ - using namespace mln; - - if (argc != 4) - return scribo::debug::usage(argv, - "Extract discontinued horizontal and vertical lines (multi-scale version)", - "input.pbm length output.ppm", - args_desc, - "A color image. Horizontal lines are in red and vertical lines in green."); - - trace::entering("main"); - - typedef image2d<bool> I; - dpoint2d none(0, 0); - - I input; - io::pbm::load(input, argv[1]); - - - // 1/1 - std::cout << "1/1" << std::endl; - I hlines = scribo::primitive::extract::lines_h_pattern(input, - atoi(argv[2]), - 3); - - - -// I vlines = scribo::primitive::extract::lines_v_pattern(input, -// atoi(argv[2]), -// 3); - - image2d<value::rgb8> out = debug::superpose(input, hlines, literal::red); -// out = debug::superpose(out, vlines, literal::green); - io::ppm::save(out, "out_1_1.ppm"); - -// I out = process(input, "out_1_1.ppm", atoi(argv[2]), 3); - - - - // 1/2 - std::cout << "1/2" << std::endl; - I input_sub2x = mln::subsampling::subsampling(input, none, 2); - I out_sub2 = process(input_sub2x, "out_1_2.ppm", atoi(argv[2]), 3, 2); - - - -// // 1/4 -// std::cout << "1/4" << std::endl; -// I input_sub4x = mln::subsampling::subsampling(input, none, 4); -// I out_sub4 = process(input_sub4x, "out_1_4.ppm", atoi(argv[2]), 3, 4); - - -// out_sub4 = world::binary_2d::enlarge(out_sub4, 2); - out_sub2 = world::binary_2d::enlarge(out_sub2, 1); - -// io::ppm::save(merge_results(input, out, out_sub2, out_sub4), argv[2]); - - I tmp = morpho::Rd::sequential(hlines, input, c8()); - io::pbm::save(tmp, "rd.pbm"); - I tmp_sub2 = morpho::Rd::sequential(out_sub2, input, c8()); - io::pbm::save(tmp_sub2, "rd_sub2.pbm"); - - out = debug::superpose(input, tmp_sub2, literal::red); - out = debug::superpose(out, tmp, literal::red); - io::ppm::save(out, "out.ppm"); - - trace::exiting("main"); -} diff --git a/scribo/src/pbm_text_in_doc.cc b/scribo/src/pbm_text_in_doc.cc new file mode 100644 index 0000000..993bbcb --- /dev/null +++ b/scribo/src/pbm_text_in_doc.cc @@ -0,0 +1,132 @@ +// Copyright (C) 2009, 2010 EPITA Research and Development Laboratory +// (LRDE) +// +// This file is part of Olena. +// +// Olena is free software: you can redistribute it and/or modify it under +// the terms of the GNU General Public License as published by the Free +// Software Foundation, version 2 of the License. +// +// Olena is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with Olena. If not, see <http://www.gnu.org/licenses/>. +// +// As a special exception, you may use this file as part of a free +// software project without restriction. Specifically, if other files +// instantiate templates or use macros or inline functions from this +// file, or you compile this file and link it with other files to produce +// an executable, this file does not by itself cause the resulting +// executable to be covered by the GNU General Public License. This +// exception does not however invalidate any other reasons why the +// executable file might be covered by the GNU General Public License. + + +#include <libgen.h> +#include <fstream> +#include <iostream> + +#include <mln/core/image/image2d.hh> +#include <mln/core/alias/neighb2d.hh> + +#include <mln/io/pbm/all.hh> + + +#include <scribo/toolchain/text_in_doc.hh> + +#include <scribo/core/line_set.hh> + +#include <scribo/debug/usage.hh> + +#include <scribo/make/debug_filename.hh> + +#include <scribo/preprocessing/crop_without_localization.hh> + +#include <scribo/io/xml/save_text_lines.hh> +#include <scribo/io/text_boxes/save.hh> + + + + +const char *args_desc[][2] = +{ + { "input.pbm", "A binary image. 'False' for object, 'True'\ +for the background." }, + { "out.txt", "Text output" }, + { "denoise_enabled", "1 enables denoising, 0 disables it. (enabled by default)" }, + { "pmin_row", "Row index of the top left corner of the Region of interest." }, + { "pmin_col", "Col index of the top left corner of the Region of interest." }, + { "pmax_row", "Row index of the bottom right corner of the Region of interest." }, + { "pmax_col", "Col index of the bottom right corner of the Region of interest." }, + { "debug_dir", "Output directory for debug image" }, + {0, 0} +}; + + +int main(int argc, char* argv[]) +{ + using namespace scribo; + using namespace mln; + + if (argc != 3 && argc != 4 && argc != 5 && argc != 8 && argc != 9) + return scribo::debug::usage(argv, + "Find text lines using left/right validation and display x-height in a binarized article.", + "input.pbm out.txt <denoise_enabled> [<pmin_row> <pmin_col> <pmax_row> <pmax_col>] <debug_dir>", + args_desc); + + bool debug = false; + + // Enable debug output. + if (argc == 5 || argc == 9) + { + scribo::make::internal::debug_filename_prefix = argv[argc - 1]; + debug = true; + } + + trace::entering("main"); + + + image2d<bool> input; + mln::io::pbm::load(input, argv[1]); + + + // Optional Cropping + point2d crop_shift = literal::origin; + if (argc >= 8) + { + mln::def::coord + minr = atoi(argv[4]), + minc = atoi(argv[5]), + maxr = atoi(argv[6]), + maxc = atoi(argv[7]); + + box2d roi = mln::make::box2d(minr, minc, maxr, maxc); + input = preprocessing::crop_without_localization(input, roi); + crop_shift = point2d(minr, minc); + + if (debug) + mln::io::pbm::save(input, + scribo::make::debug_filename("input_cropped.pbm")); + } + + bool denoise = (argc > 3 && atoi(argv[3]) != 0); + + + // Run document toolchain. + typedef image2d<scribo::def::lbl_type> L; + line_set<L> + lines = scribo::toolchain::text_in_doc(input, denoise, debug); + + + // Saving results + scribo::io::xml::save_text_lines(argv[1], lines, "out.xml"); + + // Specify shift due to previous crop. + scribo::io::text_boxes::save(lines, argv[2], crop_shift); + + + trace::exiting("main"); +} diff --git a/scribo/src/preprocessing/Makefile.am b/scribo/src/preprocessing/Makefile.am index 501233a..1f683e4 100644 --- a/scribo/src/preprocessing/Makefile.am +++ b/scribo/src/preprocessing/Makefile.am @@ -25,12 +25,34 @@ bin_PROGRAMS = \ homogeneous_contrast \ preprocess \ split_bg_fg \ - subsample \ unskew denoise_SOURCES = denoise.cc homogeneous_contrast_SOURCES = homogeneous_contrast.cc preprocess_SOURCES = preprocess.cc split_bg_fg_SOURCES = split_bg_fg.cc -subsample_SOURCES = subsample.cc unskew_SOURCES = unskew.cc + +if HAVE_MAGICKXX + + bin_PROGRAMS += to_pgm + to_pgm_SOURCES = to_pgm.cc + to_pgm_CPPFLAGS = $(AM_CPPFLAGS) \ + $(TESSERACT_CPPFLAGS) \ + `Magick++-config --cppflags` + + to_pgm_LDFLAGS = $(AM_LDFLAGS) \ + $(TESSERACT_LDFLAGS) \ + -lpthread `Magick++-config --libs` + + bin_PROGRAMS += subsample + subsample_SOURCES = subsample.cc + subsample_CPPFLAGS = $(AM_CPPFLAGS) \ + $(TESSERACT_CPPFLAGS) \ + `Magick++-config --cppflags` + + subsample_LDFLAGS = $(AM_LDFLAGS) \ + $(TESSERACT_LDFLAGS) \ + -lpthread `Magick++-config --libs` + +endif HAVE_MAGICKXX \ No newline at end of file diff --git a/scribo/src/preprocessing/subsample.cc b/scribo/src/preprocessing/subsample.cc index 388d4d9..fde31e0 100644 --- a/scribo/src/preprocessing/subsample.cc +++ b/scribo/src/preprocessing/subsample.cc @@ -25,19 +25,21 @@ #include <mln/core/image/image2d.hh> #include <mln/core/alias/neighb2d.hh> -#include <mln/io/pgm/all.hh> +#include <mln/io/magick/load.hh> +#include <mln/io/ppm/save.hh> #include <mln/subsampling/antialiased.hh> -#include <mln/value/int_u8.hh> +#include <mln/value/rgb8.hh> #include <scribo/debug/usage.hh> const char *args_desc[][2] = { - { "input.pgm", "A gray-scale image." }, + { "input.*", "An image." }, { "ratio", "Scale ratio." }, + { "output.ppm", "A color image." }, {0, 0} }; @@ -49,17 +51,17 @@ int main(int argc, char *argv[]) if (argc != 4) return scribo::debug::usage(argv, "Subsample.", - "input.pgm ratio output.pgm", + "input.* ratio output.ppm", args_desc); trace::entering("main"); - typedef image2d<value::int_u8> I; + typedef image2d<value::rgb8> I; I input; - io::pgm::load(input, argv[1]); + io::magick::load(input, argv[1]); unsigned ratio = atoi(argv[2]); - io::pgm::save(mln::subsampling::antialiased(input, ratio), argv[3]); + io::ppm::save(mln::subsampling::antialiased(input, ratio), argv[3]); trace::exiting("main"); } diff --git a/scribo/src/text/Makefile.am b/scribo/src/text/Makefile.am index aab9b19..3b3b4ae 100644 --- a/scribo/src/text/Makefile.am +++ b/scribo/src/text/Makefile.am @@ -29,15 +29,15 @@ cleantxt_SOURCES = cleantxt.cc if HAVE_TESSERACT if HAVE_TIFF - bin_PROGRAMS += recognition - recognition_SOURCES = recognition.cc - recognition_CPPFLAGS = $(AM_CPPFLAGS) \ - $(TESSERACT_CPPFLAGS) \ - $(TIFF_CPPFLAGS) - recognition_LDFLAGS = $(AM_LDFLAGS) \ - $(TESSERACT_LDFLAGS) \ - $(TIFF_LDFLAGS) \ - -lpthread + bin_PROGRAMS += pbm_lines_recognition + pbm_lines_recognition_SOURCES = pbm_lines_recognition.cc + pbm_lines_recognition_CPPFLAGS = $(AM_CPPFLAGS) \ + $(TESSERACT_CPPFLAGS) \ + $(TIFF_CPPFLAGS) + pbm_lines_recognition_LDFLAGS = $(AM_LDFLAGS) \ + $(TESSERACT_LDFLAGS) \ + $(TIFF_LDFLAGS) \ + -lpthread bin_PROGRAMS += pbm_recognition pbm_recognition_SOURCES = pbm_recognition.cc diff --git a/scribo/src/text/recognition.cc b/scribo/src/text/pbm_lines_recognition.cc similarity index 52% rename from scribo/src/text/recognition.cc rename to scribo/src/text/pbm_lines_recognition.cc index de5cafa..5aaa15b 100644 --- a/scribo/src/text/recognition.cc +++ b/scribo/src/text/pbm_lines_recognition.cc @@ -1,4 +1,4 @@ -// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE) +// Copyright (C) 2010 EPITA Research and Development Laboratory (LRDE) // // This file is part of Olena. // @@ -26,33 +26,30 @@ #include <iostream> #include <mln/core/image/image2d.hh> - -#include <mln/util/array.hh> - -#include <mln/io/txt/save.hh> #include <mln/io/pbm/load.hh> +#include <mln/value/int_u16.hh> -#include <mln/value/label_16.hh> +#include <scribo/text/recognition.hh> -#include <mln/core/alias/neighb2d.hh> +#include <scribo/debug/usage.hh> -#include <scribo/primitive/extract/objects.hh> +#include <scribo/core/component_set.hh> +#include <scribo/core/object_links.hh> +#include <scribo/core/object_groups.hh> -#include <scribo/primitive/group/apply.hh> -#include <scribo/primitive/link/with_several_left_links.hh> -#include <scribo/primitive/link/with_several_right_links.hh> -#include <scribo/primitive/group/from_double_link.hh> -#include <scribo/filter/objects_small.hh> -#include <scribo/filter/objects_thin.hh> -#include <scribo/text/recognition.hh> +#include <scribo/text/merging.hh> +#include <scribo/primitive/extract/components.hh> +#include <scribo/primitive/link/with_single_left_link_dmax_ratio.hh> +#include <scribo/primitive/group/from_single_link.hh> + +#include <scribo/io/text_boxes/save.hh> -#include <scribo/debug/usage.hh> const char *args_desc[][2] = { { "input.pbm", "A binary image. 'True' for objects, 'False'\ for the background." }, - { "out.txt", "OCR's output." }, + { "lang", "Must be set to \"eng\", \"fra\", \"deu\", \"spa\", \"ita\" (Default \"fra\")" }, {0, 0} }; @@ -63,39 +60,48 @@ int main(int argc, char* argv[]) using namespace scribo; using namespace mln; - if (argc != 3) + if (argc != 2 && argc != 3 && argc != 4) return scribo::debug::usage(argv, - "Text extraction and recognition", - "input.pbm out.txt", - args_desc, - "The text is printed on the standard output and stored in 'out.txt'"); + "Text recognition", + "input.pbm [lang] [file]", + args_desc); trace::entering("main"); image2d<bool> input; - io::pbm::load(input, argv[1]); - - /// Extract text. - typedef mln_ch_value_(image2d<bool>,value::label_16) L; - value::label_16 nbboxes; - object_image(L) - objects = scribo::primitive::extract::objects(input, c8(), nbboxes); - - /// Filter non interesting objects - objects = filter::objects_small(objects, 4); - objects = filter::objects_thin<L>(objects, 2); - - /// Group objects. - object_links<L> left_link - = primitive::link::with_several_left_links(objects, 30); - object_links<L> right_link - = primitive::link::with_several_right_links(objects, 30); + mln::io::pbm::load(input, argv[1]); + + typedef image2d<value::int_u16> L; + value::int_u16 ncomps; + component_set<L> + comps = scribo::primitive::extract::components(input, c8(), ncomps); + std::cout << "ncomps = " << ncomps << std::endl; + + object_links<L> + links = scribo::primitive::link::with_single_left_link_dmax_ratio(comps, 2); + object_groups<L> - groups = primitive::group::from_double_link(objects, left_link, right_link); + groups = scribo::primitive::group::from_single_link(links); + + line_set<L> lines(groups); + lines = text::merging(lines); + + std::string str = argv[2]; + const char *lang; + if (argc < 3 || (str != "eng" && str != "fra" + && str != "deu" && str != "spa" + && str != "ita")) + lang = "fra"; + else + lang = argv[2]; + + char *output = 0; + if (argc >= 4) + output = argv[3]; + + scribo::text::recognition(lines, lang); - objects = primitive::group::apply(objects, groups); - /// Try to recognize text in grouped objects. - scribo::text::recognition(objects, "fra", argv[2]); + scribo::io::text_boxes::save(lines, output); trace::exiting("main"); } diff --git a/scribo/src/text/pbm_recognition.cc b/scribo/src/text/pbm_recognition.cc index 70ceca1..082faef 100644 --- a/scribo/src/text/pbm_recognition.cc +++ b/scribo/src/text/pbm_recognition.cc @@ -37,6 +37,7 @@ const char *args_desc[][2] = { { "input.pbm", "A binary image. 'True' for objects, 'False'\ for the background." }, + { "lang", "Must be set to \"eng\", \"fra\", \"deu\", \"spa\", \"ita\" (Default \"fra\")" }, {0, 0} }; @@ -47,19 +48,28 @@ int main(int argc, char* argv[]) using namespace scribo; using namespace mln; - if (argc != 2) + if (argc != 2 && argc != 3) return scribo::debug::usage(argv, "Text recognition", - "input.pbm", - args_desc, - "The text is printed on the standard output"); + "input.pbm [lang]", + args_desc); trace::entering("main"); image2d<bool> input; io::pbm::load(input, argv[1]); - scribo::text::recognition(input, "fra"); + + std::string str = argv[2]; + const char *lang; + if (argc < 3 || (str != "eng" && str != "fra" + && str != "deu" && str != "spa" + && str != "ita")) + lang = "fra"; + else + lang = argv[2]; + + scribo::text::recognition(input, lang); trace::exiting("main"); } diff --git a/scribo/src/text_in_article_pbm.cc b/scribo/src/text_in_article_pbm.cc deleted file mode 100644 index 20a18e9..0000000 --- a/scribo/src/text_in_article_pbm.cc +++ /dev/null @@ -1,381 +0,0 @@ -// Copyright (C) 2009, 2010 EPITA Research and Development Laboratory -// (LRDE) -// -// This file is part of Olena. -// -// Olena is free software: you can redistribute it and/or modify it under -// the terms of the GNU General Public License as published by the Free -// Software Foundation, version 2 of the License. -// -// Olena is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU -// General Public License for more details. -// -// You should have received a copy of the GNU General Public License -// along with Olena. If not, see <http://www.gnu.org/licenses/>. -// -// As a special exception, you may use this file as part of a free -// software project without restriction. Specifically, if other files -// instantiate templates or use macros or inline functions from this -// file, or you compile this file and link it with other files to produce -// an executable, this file does not by itself cause the resulting -// executable to be covered by the GNU General Public License. This -// exception does not however invalidate any other reasons why the -// executable file might be covered by the GNU General Public License. - - -#include <libgen.h> -#include <fstream> -#include <iostream> - -#include <mln/core/image/image2d.hh> -#include <mln/core/alias/neighb2d.hh> - -#include <mln/io/pbm/all.hh> -#include <mln/io/ppm/save.hh> -#include <mln/io/dump/save.hh> - -#include <mln/literal/colors.hh> -#include <mln/value/rgb8.hh> -#include <mln/value/label_16.hh> - -#include <mln/data/paste_without_localization.hh> - -#include <scribo/core/line_set.hh> - -#include <scribo/primitive/extract/components.hh> -#include <scribo/primitive/extract/vertical_separators.hh> - -#include <scribo/primitive/remove/separators.hh> - -#include <scribo/primitive/link/merge_double_link.hh> -#include <scribo/primitive/link/internal/dmax_width_and_height.hh> -#include <scribo/primitive/link/with_single_left_link_dmax_ratio.hh> -#include <scribo/primitive/link/with_single_right_link_dmax_ratio.hh> - -#include <scribo/primitive/group/from_single_link.hh> - -#include <scribo/filter/object_links_bbox_h_ratio.hh> -#include <scribo/filter/objects_small.hh> - - -#include <scribo/debug/usage.hh> -#include <scribo/debug/save_bboxes_image.hh> -#include <scribo/debug/bboxes_enlarged_image.hh> -#include <scribo/debug/mean_and_base_lines_image.hh> -#include <scribo/debug/looks_like_a_text_line_image.hh> - -#include <scribo/make/debug_filename.hh> - -#include <scribo/text/recognition.hh> - -#include <scribo/text/merging.hh> - - -#include <scribo/preprocessing/denoise_fg.hh> - -#include <scribo/io/xml/save_text_lines.hh> - -// #include <mln/morpho/closing/structural.hh> -// #include <mln/win/rectangle2d.hh> - - -const char *args_desc[][2] = -{ - { "input.pbm", "A binary image. 'False' for object, 'True'\ -for the background." }, - { "out.txt", "Text output" }, - { "denoise", "1 enables denoising, 0 disables it. (enabled by default)" }, - { "pmin_row", "Row index of the top left corner of the Region of interest." }, - { "pmin_col", "Col index of the top left corner of the Region of interest." }, - { "pmax_row", "Row index of the bottom right corner of the Region of interest." }, - { "pmax_col", "Col index of the bottom right corner of the Region of interest." }, - { "debug_dir", "Output directory for debug image" }, - {0, 0} -}; - - -int main(int argc, char* argv[]) -{ - using namespace scribo; - using namespace mln; - - if (argc != 3 && argc != 4 && argc != 5 && argc != 8 && argc != 9) - return scribo::debug::usage(argv, - "Find text lines using left/right validation and display x-height in a binarized article.", - "input.pbm out.txt <denoise: 0|1> [<pmin_row> <pmin_col> <pmax_row> <pmax_col>] <debug_dir>", - args_desc); - - bool debug = false; - - // Enable debug output. - if (argc == 5 || argc == 9) - { - scribo::make::internal::debug_filename_prefix = argv[argc - 1]; - debug = true; - } - - trace::entering("main"); - - - image2d<bool> input; - mln::io::pbm::load(input, argv[1]); - - - // Optional Cropping - if (argc >= 8) - { - def::coord - minr = atoi(argv[4]), - minc = atoi(argv[5]), - maxr = atoi(argv[6]), - maxc = atoi(argv[7]); - - box2d roi = mln::make::box2d(minr, minc, maxr, maxc); - image2d<bool> tmp(maxr - minr + 1, maxc - minc + 1); - data::paste_without_localization(input | roi, tmp); - input = tmp; - - if (debug) - mln::io::pbm::save(input, - scribo::make::debug_filename("input_cropped.pbm")); - } - - - typedef value::label_16 V; - typedef image2d<V> L; - - - // Add whitespace separators. -// win::rectangle2d win = win::rectangle2d(151, 41); -// image2d<bool> whitespaces = morpho::closing::structural(input, win); -// logical::not_inplace(whitespaces); - - // Remove separators - std::cout << "Find vertical separators..." << std::endl; - image2d<bool> separators = primitive::extract::vertical_separators(input, 81); - - std::cout << "Remove separators..." << std::endl; - image2d<bool> input_cleaned = primitive::remove::separators(input, - separators); - -// whitespaces += separators; - - if (debug) - mln::io::pbm::save(separators, - scribo::make::debug_filename("vseparators.pbm")); -// mln::io::pbm::save(whitespaces, "separators.pbm"); - -// mln::io::pbm::save(input_cleaned, "input_no_separators.pbm"); - - // Denoise - if (argc > 3 && atoi(argv[3]) != 0) - { - std::cout << "Denoise..." << std::endl; - input_cleaned = preprocessing::denoise_fg(input_cleaned, c8(), 3); - } - -// mln::io::pbm::save(input_cleaned, "input_denoised.pbm"); - - /// Finding components. - std::cout << "Finding components..." << std::endl; - V ncomponents; - component_set<L> - components = scribo::primitive::extract::components(input_cleaned, c8(), - ncomponents); - - /// Set separator components. - components.add_separators(separators); -// components.add_separators(whitespaces); - - components = scribo::filter::components_small(components, 3); - - - /// Linking potential objects - std::cout << "Linking objects..." << std::endl; - object_links<L> left_link - = primitive::link::with_single_left_link_dmax_ratio(components, - primitive::link::internal::dmax_width_and_height(1), - anchor::MassCenter); - object_links<L> right_link - = primitive::link::with_single_right_link_dmax_ratio(components, - primitive::link::internal::dmax_width_and_height(1), - anchor::MassCenter); - - // Validating left and right links. - object_links<L> - merged_links = primitive::link::merge_double_link(left_link, right_link); - - - // Remove links if bboxes have too different sizes. - object_links<L> hratio_filtered_links - = filter::object_links_bbox_h_ratio(merged_links, 2.5f); - - -// #ifndef NOUT -// if (argc == 4) -// { -// image2d<value::rgb8> -// hratio_decision_image = scribo::debug::decision_image(input, -// merged_links, -// hratio_filtered_links); -// io::ppm::save(hratio_decision_image, -// scribo::make::debug_filename("hratio_links_decision_image.ppm")); -// } -// #endif - - - -// // FIXME: from_single_link should return a packed object_groups? -// // -// //###### - object_groups<L> - groups = primitive::group::from_single_link(hratio_filtered_links); -// value::label_16 n_groups; -// mln::fun::i2v::array<value::label_16> -// groups_packed = mln::make::relabelfun(groups, -// filtered_objects.nlabels(), -// n_groups); -// //###### - - - - // Construct a line set. - line_set<L> - lines = scribo::make::line_set(groups); - - - - //===== DEBUG ===== - - if (debug) - { - - // Bboxes image. - scribo::debug::save_bboxes_image(input, lines, - scribo::make::debug_filename("step1_bboxes.ppm")); - - // Bboxes enlarged - mln::io::ppm::save(scribo::debug::bboxes_enlarged_image(input, lines), - scribo::make::debug_filename("step1_bboxes_enlarged.ppm")); - - // Looks like a text line - mln::io::ppm::save(scribo::debug::looks_like_a_text_line_image(input, lines), - scribo::make::debug_filename("step1_looks_like_a_text_line.ppm")); - -// // Bboxes + line infos -// { -// std::ofstream file(scribo::make::debug_filename("step1_bboxes_100p.txt").c_str()); -// // std::ofstream file_50p(scribo::make::debug_filename("step1_bboxes_50p.txt").c_str()); - -// for_all_lines(l, lines) -// if (lines(l).tag() != line::Merged -// && lines(l).tag() != line::Ignored -// && lines(l).tag() != line::Pathological) -// { -// file << lines(l).bbox().pmin().row() << " " -// << lines(l).bbox().pmin().col() << " " -// << lines(l).bbox().pmax().row() << " " -// << lines(l).bbox().pmax().col() << " " -// << lines(l).card() << " " -// << lines(l).baseline() << " " -// << lines(l).x_height() << " " -// << lines(l).meanline() << " " -// << lines(l).d_height() << " " -// << lines(l).a_height() << " " -// << lines(l).char_space() << " " -// << lines(l).char_width() << std::endl; - -// // file_50p << lines(l).bbox().pmin().row() / 2 << " " -// // << lines(l).bbox().pmin().col() / 2 << " " -// // << lines(l).bbox().pmax().row() / 2 << " " -// // << lines(l).bbox().pmax().col() / 2 << std::endl; -// } - -// file.close(); -// // file_50p.close(); -// } - - - // mean and base lines. - mln::io::ppm::save(scribo::debug::mean_and_base_lines_image(input, lines), - scribo::make::debug_filename("step1_x_height.ppm")); - - } - //===== END OF DEBUG ===== - - - - - std::cout << "Merging lines..." << std::endl; - lines = scribo::text::merging(lines); - - - - //===== DEBUG ===== - - if (debug) - { - - // mean and base lines. - mln::io::ppm::save(scribo::debug::mean_and_base_lines_image(input, lines), - scribo::make::debug_filename("step2_x_height.ppm")); - - // Looks like a text line - mln::io::ppm::save(scribo::debug::looks_like_a_text_line_image(input, lines), - scribo::make::debug_filename("step2_looks_like_a_text_line.ppm")); - - // Bboxes image. - scribo::debug::save_bboxes_image(input, lines, - scribo::make::debug_filename("step2_bboxes.ppm")); - - - } - - -// { -// std::ofstream file(scribo::make::debug_filename("step2_bboxes_100p.txt").c_str()); -// // std::ofstream file_50p(scribo::make::debug_filename("step2_bboxes_50p.txt").c_str()); - -// for_all_lines(l, lines) -// if (lines(l).tag() != line::Merged -// && lines(l).tag() != line::Ignored -// && lines(l).tag() != line::Pathological) -// { -// file << lines(l).bbox().pmin().row() << " " -// << lines(l).bbox().pmin().col() << " " -// << lines(l).bbox().pmax().row() << " " -// << lines(l).bbox().pmax().col() << " " -// << lines(l).card() << " " -// << lines(l).baseline() << " " -// << lines(l).x_height() << " " -// << lines(l).meanline() << " " -// << lines(l).d_height() << " " -// << lines(l).a_height() << " " -// << lines(l).char_space() << " " -// << lines(l).char_width() << std::endl; - -// // file_50p << lines(l).bbox().pmin().row() / 2 << " " -// // << lines(l).bbox().pmin().col() / 2 << " " -// // << lines(l).bbox().pmax().row() / 2 << " " -// // << lines(l).bbox().pmax().col() / 2 << std::endl; -// } - -// file.close(); -// // file_50p.close(); -// } - - - //===== END OF DEBUG ===== - - - scribo::io::xml::save_text_lines(argv[1], lines, "out.xml"); - - - - scribo::text::recognition(lines, "fra", argv[2]); - - - trace::exiting("main"); -} diff --git a/scribo/src/text_in_doc.cc b/scribo/src/text_in_doc.cc deleted file mode 100644 index cfca7d1..0000000 --- a/scribo/src/text_in_doc.cc +++ /dev/null @@ -1,304 +0,0 @@ -// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE) -// -// This file is part of Olena. -// -// Olena is free software: you can redistribute it and/or modify it under -// the terms of the GNU General Public License as published by the Free -// Software Foundation, version 2 of the License. -// -// Olena is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU -// General Public License for more details. -// -// You should have received a copy of the GNU General Public License -// along with Olena. If not, see <http://www.gnu.org/licenses/>. -// -// As a special exception, you may use this file as part of a free -// software project without restriction. Specifically, if other files -// instantiate templates or use macros or inline functions from this -// file, or you compile this file and link it with other files to produce -// an executable, this file does not by itself cause the resulting -// executable to be covered by the GNU General Public License. This -// exception does not however invalidate any other reasons why the -// executable file might be covered by the GNU General Public License. - - -#include <libgen.h> -#include <iostream> - -#include <mln/core/image/image2d.hh> -#include <mln/core/alias/neighb2d.hh> - -#include <mln/logical/not.hh> - -#include <mln/labeling/colorize.hh> - -#include <mln/io/pbm/all.hh> -#include <mln/io/ppm/save.hh> - -#include <mln/math/min.hh> - -#include <mln/literal/colors.hh> -#include <mln/value/rgb8.hh> -#include <mln/value/label_16.hh> - -#include <mln/draw/box.hh> - -#include <mln/extension/adjust.hh> - -#include <scribo/table/erase.hh> - -#include <scribo/draw/bounding_boxes.hh> - -#include <scribo/primitive/extract/objects.hh> -#include <scribo/primitive/extract/lines_h_discontinued.hh> -#include <scribo/primitive/extract/lines_v_discontinued.hh> - -#include <scribo/primitive/link/merge_double_link.hh> -#include <scribo/primitive/link/with_single_left_link.hh> -#include <scribo/primitive/link/with_single_right_link.hh> - -#include <scribo/primitive/group/apply.hh> -#include <scribo/primitive/group/from_double_link.hh> -#include <scribo/primitive/group/from_single_link.hh> - -#include <scribo/filter/object_links_bottom_aligned.hh> -#include <scribo/filter/object_links_bbox_h_ratio.hh> -#include <scribo/filter/object_links_bbox_overlap.hh> - -#include <scribo/filter/objects_small.hh> -#include <scribo/filter/objects_thin.hh> -#include <scribo/filter/objects_thick.hh> - -#include <scribo/filter/object_groups_small.hh> - -#include <scribo/text/recognition.hh> - -#include <scribo/debug/decision_image.hh> -#include <scribo/debug/save_bboxes_image.hh> -#include <scribo/debug/save_bboxes_image.hh> -#include <scribo/debug/save_linked_bboxes_image.hh> - -#include <scribo/debug/usage.hh> - -#include <scribo/make/debug_filename.hh> - - -const char *args_desc[][2] = -{ - { "input.pbm", "A binary image. 'False' for objects, 'True'\ -for the background." }, - { "out.txt", "A text file with all the recognized text" }, - { "output_dir", "Output directory for debug image" }, - {0, 0} -}; - - -int main(int argc, char* argv[]) -{ - using namespace scribo; - using namespace mln; - - if (argc != 3 && argc != 4) - return scribo::debug::usage(argv, - "Find text in a binarized photo.", - "input.pbm out.txt <output_dir>", - args_desc, - "Debug outputs. The recognized text is printed in the standard output."); - - if (argc == 4) - scribo::make::internal::debug_filename_prefix = argv[3]; - - trace::entering("main"); - - - image2d<bool> input; - io::pbm::load(input, argv[1]); - - logical::not_inplace(input); - - - typedef image2d<value::label_16> L; - - /// Extracting vertical and horizontal lines. - std::cout << "Extracting lines..." << std::endl; - value::label_16 nhlines, nvlines; - object_image(L) - lbl_v = primitive::extract::lines_v_discontinued(input, c8(), - nvlines, 51, 8); - object_image(L) - lbl_h = primitive::extract::lines_h_discontinued(input, c8(), - nhlines, 51, 6); - - /// Reconstruct and erase tables. - std::cout << "Reconstructing and removing tables..." << std::endl; - image2d<bool> input_notables - = scribo::table::erase(input, lbl_h, lbl_v); - - - io::pbm::save(input_notables, "table_erased.ppm"); - - /// Finding objects. - std::cout << "Finding objects..." << std::endl; - value::label_16 nobjects; - object_image(L) - objects = scribo::primitive::extract::objects(input_notables, - c8(), - nobjects); - - - /// First filtering. - std::cout << "Filtering objects..." << std::endl; - object_image(L) filtered_objects - = scribo::filter::objects_small(objects, 6); - - filtered_objects - = scribo::filter::objects_thin(filtered_objects, 1); - -// filtered_objects -// = scribo::filter::objects_thick(filtered_objects, -// math::min(input.ncols(), input.nrows()) / 5); - - - - /// Linking potential objects - std::cout << "Linking objects..." << std::endl; - object_links<L> left_link - = primitive::link::with_single_left_link(filtered_objects, 30); - object_links<L> right_link - = primitive::link::with_single_right_link(filtered_objects, 30); - - -#ifndef NOUT - if (argc == 4) - { - std::cerr << "BEFORE - nobjects = " << nobjects << std::endl; - scribo::debug::save_linked_bboxes_image(input, - filtered_objects, - left_link, right_link, - literal::red, literal::cyan, - literal::yellow, - literal::green, - scribo::make::debug_filename("links.ppm")); - } -#endif - - - // Validating left and right links. - object_links<L> - merged_links = primitive::link::merge_double_link(filtered_objects, - left_link, - right_link); - - - - - // Remove links if bboxes are not aligned. - std::cout << "Filtering object links..." << std::endl; - object_links<L> bottom_filtered_links - = filter::object_links_bottom_aligned(filtered_objects, - merged_links, - 5); - - - -#ifndef NOUT - if (argc == 4) - { - image2d<value::rgb8> - bottom_decision_image = scribo::debug::decision_image(input, - merged_links, - bottom_filtered_links); - - io::ppm::save(bottom_decision_image, - scribo::make::debug_filename("bottom_links_decision_image.ppm")); - } -#endif - - - - - - // Remove links if bboxes have too different sizes. - object_links<L> hratio_filtered_links - = filter::object_links_bbox_h_ratio(filtered_objects, - bottom_filtered_links, - 0.6f); - - - - -#ifndef NOUT - if (argc == 4) - { - image2d<value::rgb8> - hratio_decision_image = scribo::debug::decision_image(input, - bottom_filtered_links, - hratio_filtered_links); - io::ppm::save(hratio_decision_image, - scribo::make::debug_filename("hratio_links_decision_image.ppm")); - } -#endif - - - - - //Remove links if bboxes overlap too much. - object_links<L> overlap_filtered_links - = filter::object_links_bbox_overlap(filtered_objects, - hratio_filtered_links, - 0.80f); - - - - -#ifndef NOUT - if (argc == 4) - { - image2d<value::rgb8> overlap_decision_image - = scribo::debug::decision_image(input, - hratio_filtered_links, - overlap_filtered_links); - io::ppm::save(overlap_decision_image, - scribo::make::debug_filename("overlap_links_decision_image.ppm")); - } -#endif - - - - - object_groups<L> - groups = primitive::group::from_single_link(filtered_objects, - overlap_filtered_links); - - -// // Remove objects part of groups with strictly less than 3 objects. -// object_groups<L> filtered_groups = filter::object_groups_small(groups, 3); - - -// #ifndef NOUT -// image2d<value::rgb8> -// decision_image = scribo::debug::decision_image(input, -// groups, filtered_groups); -// #endif - - std::cout << "Grouping objects..." << std::endl; - object_image(L) - grouped_objects = primitive::group::apply(filtered_objects, groups); - - -#ifndef NOUT - if (argc == 4) - io::ppm::save(mln::labeling::colorize(value::rgb8(), - grouped_objects, - grouped_objects.nlabels()), - scribo::make::debug_filename("out.ppm")); -#endif - - - std::cout << "Text recognition..." << std::endl; - text::recognition(grouped_objects, "fra", argv[2]); - - trace::exiting("main"); -} diff --git a/scribo/src/text_in_doc_ppm.cc b/scribo/src/text_in_doc_ppm.cc deleted file mode 100644 index 0dafc09..0000000 --- a/scribo/src/text_in_doc_ppm.cc +++ /dev/null @@ -1,322 +0,0 @@ -// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE) -// -// This file is part of Olena. -// -// Olena is free software: you can redistribute it and/or modify it under -// the terms of the GNU General Public License as published by the Free -// Software Foundation, version 2 of the License. -// -// Olena is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU -// General Public License for more details. -// -// You should have received a copy of the GNU General Public License -// along with Olena. If not, see <http://www.gnu.org/licenses/>. -// -// As a special exception, you may use this file as part of a free -// software project without restriction. Specifically, if other files -// instantiate templates or use macros or inline functions from this -// file, or you compile this file and link it with other files to produce -// an executable, this file does not by itself cause the resulting -// executable to be covered by the GNU General Public License. This -// exception does not however invalidate any other reasons why the -// executable file might be covered by the GNU General Public License. - - -#include <libgen.h> -#include <iostream> - -#include <mln/core/image/image2d.hh> -#include <mln/core/alias/neighb2d.hh> - -#include <mln/logical/not.hh> - -#include <mln/labeling/colorize.hh> - -#include <mln/io/pbm/all.hh> -#include <mln/io/ppm/save.hh> - -#include <mln/math/min.hh> - -#include <mln/literal/colors.hh> -#include <mln/value/rgb8.hh> -#include <mln/value/label_16.hh> - -#include <mln/draw/box.hh> - -#include <mln/extension/adjust.hh> - -#include <scribo/binarization/sauvola.hh> - -#include <scribo/table/erase.hh> - -#include <scribo/draw/bounding_boxes.hh> - -#include <scribo/primitive/extract/objects.hh> -#include <scribo/primitive/extract/lines_h_discontinued.hh> -#include <scribo/primitive/extract/lines_v_discontinued.hh> - -#include <scribo/primitive/link/merge_double_link.hh> -#include <scribo/primitive/link/with_single_left_link.hh> -#include <scribo/primitive/link/with_single_right_link.hh> - -#include <scribo/primitive/group/apply.hh> -#include <scribo/primitive/group/from_double_link.hh> -#include <scribo/primitive/group/from_single_link.hh> - -#include <scribo/filter/object_links_bottom_aligned.hh> -#include <scribo/filter/object_links_bbox_h_ratio.hh> -#include <scribo/filter/object_links_bbox_overlap.hh> - -#include <scribo/filter/objects_small.hh> -#include <scribo/filter/objects_thin.hh> -#include <scribo/filter/objects_thick.hh> - -#include <scribo/filter/object_groups_small.hh> - -#include <scribo/text/recognition.hh> - -#include <scribo/debug/decision_image.hh> -#include <scribo/debug/save_bboxes_image.hh> -#include <scribo/debug/save_bboxes_image.hh> -#include <scribo/debug/save_linked_bboxes_image.hh> - -#include <scribo/debug/usage.hh> - -#include <scribo/preprocessing/split_bg_fg.hh> - -#include <scribo/make/debug_filename.hh> - - -const char *args_desc[][2] = -{ - { "input.ppm", "A color image." }, - { "out.txt", "A text file with all the recognized text" }, - { "output_dir", "Output directory for debug image" }, - {0, 0} -}; - - -int main(int argc, char* argv[]) -{ - using namespace scribo; - using namespace mln; - - if (argc != 3 && argc != 4) - return scribo::debug::usage(argv, - "Find text in a color document.", - "input.ppm out.txt <output_dir>", - args_desc, - "Debug outputs. The recognized text is printed in the standard output."); - - if (argc == 4) - scribo::make::internal::debug_filename_prefix = argv[3]; - - trace::entering("main"); - - - image2d<value::rgb8> input_rgb; - io::ppm::load(input_rgb, argv[1]); - - - unsigned lambda; - if (argc == 5) - lambda = atoi(argv[4]); - else - lambda = 1.2 * (input_rgb.nrows() + input_rgb.ncols()); - - // Extract foreground - std::cout << "Extracting foreground..." << std::endl; - image2d<value::rgb8> - fg = preprocessing::split_bg_fg(input_rgb, - lambda, - 32).second(); - - // Binarize foreground to use it in the processing chain. - std::cout << "Binarizing foreground..." << std::endl; - image2d<bool> input = scribo::binarization::sauvola(fg); - - - typedef image2d<value::label_16> L; - - /// Extracting vertical and horizontal lines. - std::cout << "Extracting lines..." << std::endl; - value::label_16 nhlines, nvlines; - object_image(L) - lbl_v = primitive::extract::lines_v_discontinued(input, c8(), - nvlines, 51, 8); - object_image(L) - lbl_h = primitive::extract::lines_h_discontinued(input, c8(), - nhlines, 51, 6); - - /// Reconstruct and erase tables. - std::cout << "Reconstructing and removing tables..." << std::endl; - image2d<bool> input_notables - = scribo::table::erase(input, lbl_h, lbl_v); - - - io::pbm::save(input_notables, "table_erased.ppm"); - - /// Finding objects. - std::cout << "Finding objects..." << std::endl; - value::label_16 nobjects; - object_image(L) - objects = scribo::primitive::extract::objects(input_notables, - c8(), - nobjects); - - - /// First filtering. - std::cout << "Filtering objects..." << std::endl; - object_image(L) filtered_objects - = scribo::filter::objects_small(objects, 6); - - filtered_objects - = scribo::filter::objects_thin(filtered_objects, 1); - -// filtered_objects -// = scribo::filter::objects_thick(filtered_objects, -// math::min(input.ncols(), input.nrows()) / 5); - - - - /// Linking potential objects - std::cout << "Linking objects..." << std::endl; - object_links<L> left_link - = primitive::link::with_single_left_link(filtered_objects, 30); - object_links<L> right_link - = primitive::link::with_single_right_link(filtered_objects, 30); - - -#ifndef NOUT - if (argc == 4) - { - std::cerr << "BEFORE - nobjects = " << nobjects << std::endl; - scribo::debug::save_linked_bboxes_image(input, - filtered_objects, - left_link, right_link, - literal::red, literal::cyan, - literal::yellow, - literal::green, - scribo::make::debug_filename("links.ppm")); - } -#endif - - - // Validating left and right links. - object_links<L> - merged_links = primitive::link::merge_double_link(filtered_objects, - left_link, - right_link); - - - - - // Remove links if bboxes are not aligned. - std::cout << "Filtering object links..." << std::endl; - object_links<L> bottom_filtered_links - = filter::object_links_bottom_aligned(filtered_objects, - merged_links, - 5); - - - -#ifndef NOUT - if (argc == 4) - { - image2d<value::rgb8> - bottom_decision_image = scribo::debug::decision_image(input, - merged_links, - bottom_filtered_links); - - io::ppm::save(bottom_decision_image, - scribo::make::debug_filename("bottom_links_decision_image.ppm")); - } -#endif - - - - - -// // Remove links if bboxes have too different sizes. -// object_links<L> hratio_filtered_links -// = filter::object_links_bbox_h_ratio(filtered_objects, -// bottom_filtered_links, -// 0.6f); - - - - -// #ifndef NOUT -// if (argc == 4) -// { -// image2d<value::rgb8> -// hratio_decision_image = scribo::debug::decision_image(input, -// bottom_filtered_links, -// hratio_filtered_links); -// io::ppm::save(hratio_decision_image, -// scribo::make::debug_filename("hratio_links_decision_image.ppm")); -// } -// #endif - - - - - //Remove links if bboxes overlap too much. - object_links<L> overlap_filtered_links - = filter::object_links_bbox_overlap(filtered_objects, - bottom_filtered_links, - 0.80f); - - - - -#ifndef NOUT - if (argc == 4) - { - image2d<value::rgb8> overlap_decision_image - = scribo::debug::decision_image(input, - bottom_filtered_links, - overlap_filtered_links); - io::ppm::save(overlap_decision_image, - scribo::make::debug_filename("overlap_links_decision_image.ppm")); - } -#endif - - - - - object_groups<L> - groups = primitive::group::from_single_link(filtered_objects, - overlap_filtered_links); - - -// // Remove objects part of groups with strictly less than 3 objects. -// object_groups<L> filtered_groups = filter::object_groups_small(groups, 3); - - -// #ifndef NOUT -// image2d<value::rgb8> -// decision_image = scribo::debug::decision_image(input, -// groups, filtered_groups); -// #endif - - std::cout << "Grouping objects..." << std::endl; - object_image(L) - grouped_objects = primitive::group::apply(filtered_objects, groups); - - -#ifndef NOUT - if (argc == 4) - io::ppm::save(mln::labeling::colorize(value::rgb8(), - grouped_objects, - grouped_objects.nlabels()), - scribo::make::debug_filename("out.ppm")); -#endif - - std::cout << "Text recognition..." << std::endl; - text::recognition(grouped_objects, "fra", argv[2]); - - trace::exiting("main"); -} diff --git a/scribo/src/text_in_article_preprocess.cc b/scribo/src/text_in_doc_preprocess.cc similarity index 84% rename from scribo/src/text_in_article_preprocess.cc rename to scribo/src/text_in_doc_preprocess.cc index bf70c7f..27ed9ed 100644 --- a/scribo/src/text_in_article_preprocess.cc +++ b/scribo/src/text_in_doc_preprocess.cc @@ -41,8 +41,11 @@ #include <scribo/debug/usage.hh> #include <scribo/preprocessing/split_bg_fg.hh> +#include <scribo/preprocessing/deskew.hh> +#include <mln/io/pgm/all.hh> + const char *args_desc[][2] = { @@ -81,11 +84,20 @@ int main(int argc, char* argv[]) input_rgb = preprocessing::split_bg_fg(input_rgb, lambda, 32).second(); } + // Convert to Gray level image. + image2d<value::int_u8> + input_gl = data::transform(input_rgb, mln::fun::v2v::rgb_to_int_u<8>()); + + + // Deskewing + std::cout << "Deskew if needed..." << std::endl; + input_gl = preprocessing::deskew(input_gl); + // Binarize foreground to use it in the processing chain. std::cout << "Binarizing foreground..." << std::endl; - image2d<bool> input = scribo::binarization::sauvola_ms(input_rgb, 101, 3); + image2d<bool> input_bin = scribo::binarization::sauvola_ms(input_gl, 101, 3); - logical::not_inplace(input); + logical::not_inplace(input_bin); - mln::io::pbm::save(input, argv[2]); + mln::io::pbm::save(input_bin, argv[2]); } diff --git a/scribo/src/text_in_photo.cc b/scribo/src/text_in_photo.cc deleted file mode 100644 index 5426dce..0000000 --- a/scribo/src/text_in_photo.cc +++ /dev/null @@ -1,250 +0,0 @@ -// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE) -// -// This file is part of Olena. -// -// Olena is free software: you can redistribute it and/or modify it under -// the terms of the GNU General Public License as published by the Free -// Software Foundation, version 2 of the License. -// -// Olena is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU -// General Public License for more details. -// -// You should have received a copy of the GNU General Public License -// along with Olena. If not, see <http://www.gnu.org/licenses/>. -// -// As a special exception, you may use this file as part of a free -// software project without restriction. Specifically, if other files -// instantiate templates or use macros or inline functions from this -// file, or you compile this file and link it with other files to produce -// an executable, this file does not by itself cause the resulting -// executable to be covered by the GNU General Public License. This -// exception does not however invalidate any other reasons why the -// executable file might be covered by the GNU General Public License. - -#include <libgen.h> -#include <iostream> - -#include <mln/core/image/image2d.hh> -#include <mln/core/alias/neighb2d.hh> - -#include <mln/labeling/colorize.hh> - -#include <mln/io/pbm/all.hh> -#include <mln/io/ppm/save.hh> - -#include <mln/math/min.hh> - -#include <mln/literal/colors.hh> -#include <mln/value/rgb8.hh> -#include <mln/value/label_16.hh> - -#include <mln/draw/box.hh> - - -#include <scribo/draw/bounding_boxes.hh> - -#include <scribo/primitive/extract/objects.hh> - -#include <scribo/primitive/link/merge_double_link.hh> -#include <scribo/primitive/link/with_single_left_link.hh> -#include <scribo/primitive/link/with_single_right_link.hh> - -#include <scribo/primitive/group/apply.hh> -#include <scribo/primitive/group/from_double_link.hh> -#include <scribo/primitive/group/from_single_link.hh> - -#include <scribo/filter/object_links_bbox_h_ratio.hh> -#include <scribo/filter/object_links_bbox_overlap.hh> - -#include <scribo/filter/objects_small.hh> -#include <scribo/filter/objects_thin.hh> -#include <scribo/filter/objects_thick.hh> - -#include <scribo/filter/object_groups_small.hh> - -#include <scribo/debug/decision_image.hh> -#include <scribo/debug/save_bboxes_image.hh> -#include <scribo/debug/save_bboxes_image.hh> -#include <scribo/debug/save_linked_bboxes_image.hh> - -#include <scribo/debug/usage.hh> - -#include <scribo/make/debug_filename.hh> - -const char *args_desc[][2] = -{ - { "input.pbm", "A binary image. 'True' for objects, 'False'\ -for the background." }, - {0, 0} -}; - - -int main(int argc, char* argv[]) -{ - using namespace scribo; - using namespace mln; - - if (argc != 3) - return scribo::debug::usage(argv, - "Find text in a binarized photo.", - "input.pbm output_dir", - args_desc, - "A color image where the text is \ -highlighted."); - - scribo::make::internal::debug_filename_prefix = argv[2]; - - trace::entering("main"); - - image2d<bool> input; - io::pbm::load(input, argv[1]); - - typedef image2d<value::label_16> L; - - /// Finding objects. - value::label_16 nobjects; - object_image(L) - objects = scribo::primitive::extract::objects(input, c8(), nobjects); - - /// First filtering. - object_image(L) filtered_objects - = scribo::filter::objects_small(objects, 6); - - filtered_objects - = scribo::filter::objects_thin(filtered_objects, 1); - -// filtered_objects -// = scribo::filter::objects_thick(filtered_objects, -// math::min(input.ncols(), input.nrows()) / 5); - - - /// Grouping potential objects - object_links<L> left_link - = primitive::link::with_single_left_link(filtered_objects, 30); - object_links<L> right_link - = primitive::link::with_single_right_link(filtered_objects, 30); - - - - - -#ifndef NOUT - std::cout << "BEFORE - nobjects = " << nobjects << std::endl; - scribo::debug::save_linked_bboxes_image(input, - filtered_objects, - left_link, right_link, - literal::red, literal::cyan, - literal::yellow, - literal::green, - scribo::make::debug_filename("links.ppm")); -#endif - - - - // Validating left and right links. - object_links<L> - merged_links = primitive::link::merge_double_link(filtered_objects, - left_link, - right_link); - // Remove links if bboxes have too different sizes. - object_links<L> - hratio_filtered_links = filter::object_links_bbox_h_ratio(filtered_objects, - merged_links, - 0.7f); - - - - -#ifndef NOUT - image2d<value::rgb8> - hratio_decision_image = scribo::debug::decision_image(input, - merged_links, - hratio_filtered_links); - io::ppm::save(hratio_decision_image, - scribo::make::debug_filename("hratio_links_decision_image.ppm")); -#endif - - - - - //Remove links if bboxes overlap too much. - object_links<L> overlap_filtered_links - = filter::object_links_bbox_overlap(filtered_objects, - hratio_filtered_links, - 0.80f); - - - - -#ifndef NOUT - image2d<value::rgb8> overlap_decision_image - = scribo::debug::decision_image(input, - hratio_filtered_links, - overlap_filtered_links); - io::ppm::save(overlap_decision_image, - scribo::make::debug_filename("overlap_links_decision_image.ppm")); -#endif - - - - - object_groups<L> - groups = primitive::group::from_single_link(filtered_objects, - overlap_filtered_links); - - - // Remove objects part of groups with strictly less than 3 objects. - object_groups<L> filtered_groups = filter::object_groups_small(groups, 3); - - - - -#ifndef NOUT - image2d<value::rgb8> decision_image; - if (argc == 4) - decision_image = scribo::debug::decision_image(input, - groups, filtered_groups); -#endif - - - - /// Apply grouping in the object image. - object_image(L) - grouped_objects = primitive::group::apply(filtered_objects, - filtered_groups); - - - /// Objects have been grouped. We try to link groups together. - /// This time a single link is enough since non-wanted objects have - /// been removed. - left_link - = primitive::link::with_single_left_link(grouped_objects, 30); - - /// Grouping groups. - groups = primitive::group::from_single_link(grouped_objects, left_link); - - grouped_objects = primitive::group::apply(grouped_objects, groups); - - - - -#ifndef NOUT - scribo::draw::bounding_boxes(decision_image, grouped_objects, literal::blue); - io::ppm::save(decision_image, scribo::make::debug_filename("decision_image.ppm")); - - std::cout << "AFTER - nobjects = " << grouped_objects.nlabels() << std::endl; -#endif - - - - io::ppm::save(mln::labeling::colorize(value::rgb8(), - grouped_objects, - grouped_objects.nlabels()), -// "/dev/null"); - scribo::make::debug_filename("out.ppm")); - - - trace::exiting("main"); -} diff --git a/scribo/src/text_in_photo_invert.cc b/scribo/src/text_in_photo_invert.cc deleted file mode 100644 index 71baee6..0000000 --- a/scribo/src/text_in_photo_invert.cc +++ /dev/null @@ -1,245 +0,0 @@ -// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE) -// -// This file is part of Olena. -// -// Olena is free software: you can redistribute it and/or modify it under -// the terms of the GNU General Public License as published by the Free -// Software Foundation, version 2 of the License. -// -// Olena is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU -// General Public License for more details. -// -// You should have received a copy of the GNU General Public License -// along with Olena. If not, see <http://www.gnu.org/licenses/>. -// -// As a special exception, you may use this file as part of a free -// software project without restriction. Specifically, if other files -// instantiate templates or use macros or inline functions from this -// file, or you compile this file and link it with other files to produce -// an executable, this file does not by itself cause the resulting -// executable to be covered by the GNU General Public License. This -// exception does not however invalidate any other reasons why the -// executable file might be covered by the GNU General Public License. - -#include <libgen.h> -#include <iostream> - -#include <mln/core/image/image2d.hh> -#include <mln/core/alias/neighb2d.hh> - -#include <mln/labeling/colorize.hh> - -#include <mln/io/pbm/all.hh> -#include <mln/io/pgm/all.hh> -#include <mln/io/ppm/all.hh> - -#include <mln/math/min.hh> - -#include <mln/literal/colors.hh> -#include <mln/value/rgb8.hh> -#include <mln/value/label_16.hh> - -#include <mln/draw/box.hh> - -#include <mln/logical/not.hh> - -#include <mln/labeling/superpose.hh> - -#include <scribo/primitive/extract/objects.hh> - -#include <scribo/primitive/link/with_single_left_link.hh> -#include <scribo/primitive/link/with_single_right_link.hh> - -#include <scribo/primitive/group/apply.hh> -#include <scribo/primitive/group/from_double_link.hh> -#include <scribo/primitive/group/from_single_link.hh> - -#include <scribo/filter/objects_small.hh> -#include <scribo/filter/objects_thin.hh> -#include <scribo/filter/objects_thick.hh> -#include <scribo/filter/object_groups_small.hh> - -#include <scribo/make/debug_filename.hh> - -#include <scribo/debug/decision_image.hh> -#include <scribo/debug/save_bboxes_image.hh> -#include <scribo/debug/save_linked_bboxes_image.hh> - -#include <scribo/debug/usage.hh> - -const char *args_desc[][2] = -{ - { "input.pbm", "A binary image. 'True' for objects, 'False'\ -for the background." }, - { "input_i.pbm", "A binary image. 'True' for objects, 'False'\ -for the background." }, - {0, 0} -}; - - - -namespace mln -{ - - template <typename V> - object_image(image2d<value::label_16>) - run(const image2d<V>& input) - { - using namespace scribo; - - typedef image2d<value::label_16> L; - - /// Finding objects. - value::label_16 nobjects; - object_image(L) - objects = scribo::primitive::extract::objects(input, c8(), nobjects); - - - /// First filtering. - object_image(L) filtered_objects - = scribo::filter::objects_small(objects, 6); - - filtered_objects - = scribo::filter::objects_thin(filtered_objects, 1); - - filtered_objects - = scribo::filter::objects_thick(filtered_objects, - math::min(input.ncols(), input.nrows()) / 6); - - - /// Finding links between potential objects - object_links<L> left_link - = primitive::link::with_single_left_link(filtered_objects, 30); - object_links<L> right_link - = primitive::link::with_single_right_link(filtered_objects, 30); - - - - std::cout << "BEFORE - nobjects = " << nobjects << std::endl; - scribo::debug::save_linked_bboxes_image(input, - filtered_objects, - left_link, right_link, - literal::red, literal::cyan, - literal::yellow, - literal::green, - scribo::make::debug_filename("links.ppm")); - - - - - // Trying to group objects - object_groups<L> - groups = primitive::group::from_double_link(filtered_objects, - left_link, right_link); - - // Remove objects part of groups with strictly less than 3 objects. - object_groups<L> filtered_groups = filter::object_groups_small(groups, 3); - -#ifndef NOUT - image2d<value::rgb8> - decision_image = scribo::debug::decision_image(input, - groups, filtered_groups); -#endif - - - /// Apply grouping in the object image. - object_image(L) - grouped_objects = primitive::group::apply(filtered_objects, - filtered_groups); - - - // Objects have been removed we need to update object links again. - // This time a single link is enough since non-wanted objects have - // been removed. - left_link - = primitive::link::with_single_left_link(filtered_objects, 30); - - - - // Grouping objects again. - groups = primitive::group::from_single_link(filtered_objects, left_link); - - grouped_objects = primitive::group::apply(filtered_objects, groups); - -#ifndef NOUT - for (unsigned i = 1; i <= grouped_objects.nlabels(); ++i) - mln::draw::box(decision_image, grouped_objects.bbox(i), literal::blue); - - io::ppm::save(decision_image, scribo::make::debug_filename("decision_image.ppm")); -#endif // ! NOUT - - - std::cout << "AFTER - nobjects = " << grouped_objects.nlabels() << std::endl; -// - - return grouped_objects; - } - - -} - - - -int main(int argc, char* argv[]) -{ - using namespace scribo; - using namespace mln; - - if (argc != 4) - return scribo::debug::usage(argv, - "Find text in a binarized photo.", - "input.pbm input_i.pbm output_dir", - args_desc, - "Several color images"); - - - scribo::make::internal::debug_filename_prefix = argv[3]; - - - trace::entering("main"); - - image2d<bool> input; - io::pbm::load(input, argv[1]); - - typedef image2d<value::label_16> L; - - value::label_16 new_nlabels; - - object_image(L) out_1 = run(input); - -#ifndef NOUT - io::ppm::save(mln::labeling::colorize(value::rgb8(), - out_1, - new_nlabels), - scribo::make::debug_filename("out_normal.ppm")); -#endif // ! NOUT - - - image2d<bool> input_i; - io::pbm::load(input_i, argv[2]); - - object_image(L) out_2 = run(input_i); - -#ifndef NOUT - io::ppm::save(mln::labeling::colorize(value::rgb8(), - out_2, - new_nlabels), - scribo::make::debug_filename("out_inverted.ppm")); -#endif // ! NOUT - - - L out = labeling::superpose(out_1, out_1.nlabels(), - out_2, out_2.nlabels(), - new_nlabels); - - - io::ppm::save(mln::labeling::colorize(value::rgb8(), - out, - new_nlabels), - scribo::make::debug_filename("out_combined.ppm")); - - - trace::exiting("main"); -} diff --git a/scribo/src/text_in_photo_pbm_fast.cc b/scribo/src/text_in_photo_pbm_fast.cc deleted file mode 100644 index 6b27bbc..0000000 --- a/scribo/src/text_in_photo_pbm_fast.cc +++ /dev/null @@ -1,441 +0,0 @@ -// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE) -// -// This file is part of Olena. -// -// Olena is free software: you can redistribute it and/or modify it under -// the terms of the GNU General Public License as published by the Free -// Software Foundation, version 2 of the License. -// -// Olena is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU -// General Public License for more details. -// -// You should have received a copy of the GNU General Public License -// along with Olena. If not, see <http://www.gnu.org/licenses/>. -// -// As a special exception, you may use this file as part of a free -// software project without restriction. Specifically, if other files -// instantiate templates or use macros or inline functions from this -// file, or you compile this file and link it with other files to produce -// an executable, this file does not by itself cause the resulting -// executable to be covered by the GNU General Public License. This -// exception does not however invalidate any other reasons why the -// executable file might be covered by the GNU General Public License. - -#include <libgen.h> -#include <iostream> - -#include <mln/core/image/image2d.hh> -#include <mln/core/image/imorph/tr_image.hh> -#include <mln/core/alias/neighb2d.hh> - -#include <mln/labeling/colorize.hh> - -#include <mln/data/stretch.hh> - -#include <mln/io/pbm/all.hh> -#include <mln/io/ppm/save.hh> - -#include <mln/math/min.hh> - -#include <mln/logical/not.hh> - -#include <mln/literal/colors.hh> -#include <mln/value/rgb8.hh> -#include <mln/value/label_16.hh> - -#include <mln/fun/v2v/rgb_to_int_u.hh> - -#include <mln/data/wrap.hh> - -#include <mln/draw/box.hh> - -#include <mln/geom/translate.hh> - -#include <scribo/draw/bounding_boxes.hh> - -#include <scribo/binarization/sauvola_ms.hh> -#include <scribo/binarization/sauvola.hh> - -#include <scribo/primitive/extract/components.hh> - -#include <scribo/primitive/link/merge_double_link.hh> -#include <scribo/primitive/link/with_single_left_link.hh> -#include <scribo/primitive/link/with_single_right_link.hh> - -#include <scribo/primitive/group/apply.hh> -#include <scribo/primitive/group/from_double_link.hh> -#include <scribo/primitive/group/from_single_link.hh> - -#include <scribo/filter/objects_with_holes.hh> -#include <scribo/filter/object_links_bbox_h_ratio.hh> -#include <scribo/filter/object_links_bbox_overlap.hh> - -#include <scribo/filter/common/objects_photo.hh> - -#include <scribo/filter/object_groups_small.hh> -#include <scribo/filter/object_groups_v_thickness.hh> - -#include <scribo/debug/decision_image.hh> -#include <scribo/debug/save_bboxes_image.hh> -#include <scribo/debug/save_linked_bboxes_image.hh> - -#include <scribo/debug/usage.hh> - -#include <scribo/preprocessing/split_bg_fg.hh> - -#include <scribo/make/debug_filename.hh> - -#include <mln/util/timer.hh> -#include <mln/core/var.hh> - - -#include <scribo/src/afp/components.hh> -#include <scribo/src/afp/link.hh> -#include <scribo/src/afp/regroup.hh> - -#include <scribo/core/line_set.hh> - -const char *args_desc[][2] = -{ - { "input.ppm", "A color image." }, - { "debug_output_dir", "Directory were debug images will be saved" }, - { "lambda", "Lambda value used for foreground extraction" }, - {0, 0} -}; - - -namespace mln -{ - - struct mask_non_text : Function_v2v<mask_non_text> - { - typedef value::rgb8 result; - typedef image2d<bool> I; - - mask_non_text(const image2d<bool>& mask) - : mask_(mask), p_(mask_) - { - p_.start(); - } - - result operator()(const result& v) const - { - bool b = p_.val(); - p_.next(); - if (!b) - return v / 2; - else - return v; - - } - - I mask_; - mutable mln_pixter_(I) p_; - }; - - - template <typename I, typename L> - mln_concrete(I) - compute_highlight_image(const I& input_rgb, - const scribo::component_set<L>& components) - { - mln_ch_value(I, bool) mask; - initialize(mask, input_rgb); - data::fill(mask, false); - - for_all_comps(i, components) - if (components(i).is_valid()) - data::fill((mask | components(i).bbox()).rw(), true); - - mask_non_text f(mask); - mln_concrete(I) output = data::transform(input_rgb, f); - - for_all_comps(i, components) - if (components(i).is_valid()) - mln::draw::box(output, components(i).bbox(), literal::red); - - return output; - } - - template <typename I, typename L> - mln_concrete(I) - compute_text_image(const I& input_rgb, - const scribo::component_set<L>& grouped_objects) - { - unsigned shift = 5; - float height = 1, width = 0; - for_all_comps(i, grouped_objects) - if (grouped_objects(i).is_valid()) - { - height += grouped_objects(i).bbox().nrows() + shift; - width = math::max(static_cast<float>(grouped_objects(i).bbox().ncols()), - width); - } - if (width == 0) - width = 1; - - I output(height, width); - data::fill(output, literal::black); - - algebra::vec<2, float> dv; - dv[0] = 0; - dv[1] = 0; - for_all_comps(i, grouped_objects) - if (grouped_objects(i).is_valid()) - { - mln_VAR(tmp, duplicate(input_rgb | grouped_objects(i).bbox())); - - typedef fun::x2x::translation<mln_site_(I)::dim, float> trans_t; - trans_t trans(dv - grouped_objects(i).bbox().pmin().to_vec()); - - mln_domain(I) - tr_box(grouped_objects(i).bbox().pmin().to_vec() + trans.t(), - grouped_objects(i).bbox().pmax().to_vec() + trans.t()); - - tr_image<mln_domain(I), tmp_t, trans_t> tr_ima(tr_box, tmp, trans); - - data::paste(tr_ima, output); - dv[0] += grouped_objects(i).bbox().nrows() + shift; - } - - return output; - } - -} // end of namespace mln - - -int main(int argc, char* argv[]) -{ - using namespace scribo; - using namespace mln; - - if (argc < 3 || argc > 8) - return scribo::debug::usage(argv, - "Find text in a photo.\n\n\ -Common usage: ./pbm_text_in_photo input.pbm output.ppm 1 1 1", - "input.pbm output.ppm <Bg comp filter enabled> <small group filter enabled> <thin group filter enabled> [debug_output_dir] [lambda]", - args_desc, - "A color image where the text is highlighted."); - - std::string out_base_dir; - bool debug = false; - if (argc > 6) - { - scribo::make::internal::debug_filename_prefix = argv[6]; - debug = true; - out_base_dir = argv[6]; - } - - trace::entering("main"); - - image2d<bool> input; - io::pbm::load(input, argv[1]); - - - mln::util::timer timer_, global_t_; - float t_; - - - - typedef image2d<value::label_16> L; - - /// Finding components. - timer_.restart(); - - typedef component_set<L> Obj; - Obj filtered_components; - - { - mln::util::array<std::pair<box2d, std::pair<point2d, unsigned> > > attribs; - - value::label_16 ncomponents; - L components = extract_components(input, ncomponents, attribs); - - filtered_components = Obj(components, ncomponents, attribs); - } - - t_ = timer_; - std::cout << "Object extracted " << t_ << " - " << filtered_components.nelements() << " components" << std::endl; - - /// linking potential components - timer_.restart(); - mln::util::couple<object_links<L>, object_links<L> > - links = primitive::link::left_right(filtered_components); - - object_links<L>& left_link = links.first(); - object_links<L>& right_link = links.second(); - - t_ = timer_; - std::cout << "Link done " << t_ << std::endl; - - - - // Validating left and right links. - timer_.restart(); - object_links<L> - merged_links = primitive::link::merge_double_link(left_link, right_link); - t_ = timer_; - std::cout << "Right/Left Validation. " << t_ << std::endl; - - - // Remove links if bboxes have too different sizes. - timer_.restart(); - object_links<L> - hratio_filtered_links = filter::object_links_bbox_h_ratio(merged_links, - 1.50f); - - //Remove links if bboxes overlap too much. - object_links<L> overlap_filtered_links - = filter::object_links_bbox_overlap(hratio_filtered_links, 0.80f); - - t_ = timer_; - std::cout << "Components links filtered. " << t_ << std::endl; - - timer_.restart(); - object_groups<L> - groups = primitive::group::from_single_link(overlap_filtered_links); - - - -// Apply grouping in a temporary image (for debug purpose). - - t_ = timer_; - std::cout << "Components grouped. " << t_ << std::endl; - - std::cout << "Filtering groups..." << std::endl; - - mln::util::timer g_timer; - - timer_.restart(); - // Remove components part of groups with strictly less than 3 components. - - g_timer.start(); - object_groups<L> filtered_small_groups; - if (argc > 4 && atoi(argv[4]) != 0) - { - std::cout << "** Using group too small" << std::endl; - filtered_small_groups = filter::object_groups_small(groups, 3); - } - else - filtered_small_groups = groups; - t_ = g_timer; - std::cout << "Small groups removed " << t_ << std::endl; - - - - // Remove components part of groups having a mean thickness lower than 8. - g_timer.restart(); - object_groups<L> filtered_thin_groups; - if (argc > 5 && atoi(argv[5]) != 0) - { - std::cout << "** Using group too thin" << std::endl; - filtered_thin_groups - = filter::object_groups_v_thickness(filtered_small_groups, 8); - } - else - filtered_thin_groups = filtered_small_groups; - t_ = g_timer; - std::cout << "Groups too thin " << t_ << std::endl; - - - - /// Apply grouping in the object image. - g_timer.restart(); - -// groups = primitive::group::regroup_left(filtered_components, -// filtered_thin_groups, -// 30); - - component_set<L> - grouped_components = primitive::group::apply(filtered_thin_groups); - - t_ = g_timer; - std::cout << "Group applied to object image " << t_ << std::endl; - - /// Components have been grouped. We try to link groups together. - /// This time a single link is enough since non-wanted components have - /// been removed. - g_timer.restart(); - - left_link = primitive::link::left(grouped_components, 30); - - - /// Grouping groups. - groups = primitive::group::from_single_link(left_link); - - - grouped_components = primitive::group::apply(groups); - - t_ = g_timer; - std::cout << "Link and group again " << t_ << std::endl; - - timer_.stop(); - - timer_.resume(); - - g_timer.restart(); - - - /// Filter grouped objects not having enough background components. - if (argc > 3 && atoi(argv[3]) != 0) - { - std::cout << "** Using objects_with_two_holes" << std::endl; - grouped_components = scribo::filter::components_with_two_holes(grouped_components, 5); - t_ = g_timer; - std::cout << "Objects_with_holes " << t_ << std::endl; - } - - - t_ = timer_; - std::cout << "Objects groups filtered. " << t_ << std::endl; - - - t_ = global_t_; - std::cout << "*** Result computed " << t_ << std::endl; - - timer_.restart(); - std::cout << "Saving result..." << std::endl; - io::ppm::save(mln::labeling::colorize(value::rgb8(), - grouped_components.labeled_image(), - grouped_components.nelements()), - argv[2]); - - - scribo::line_set<L> - lines = scribo::line_set<L>(filtered_thin_groups); - - - // Bboxes + line infos - { - std::ofstream file(scribo::make::debug_filename("text_bboxes.txt").c_str()); - - for_all_lines(l, lines) - if (lines(l).tag() != line::Merged - && lines(l).tag() != line::Ignored - && lines(l).tag() != line::Pathological) - { - file << lines(l).bbox().pmin().row() << " " - << lines(l).bbox().pmin().col() << " " - << lines(l).bbox().pmax().row() << " " - << lines(l).bbox().pmax().col() -// << " " -// << lines(l).card() << " " -// << lines(l).baseline() << " " -// << lines(l).x_height() << " " -// << lines(l).median() << " " -// << lines(l).d_height() << " " -// << lines(l).a_height() << " " -// << lines(l).char_space() << " " -// << lines(l).char_width() - << std::endl; - } - - file.close(); - } - - - trace::exiting("main"); - return grouped_components.nelements() != 0; -} diff --git a/scribo/src/text_in_photo_ppm.cc b/scribo/src/text_in_photo_ppm.cc deleted file mode 100644 index 9027ab0..0000000 --- a/scribo/src/text_in_photo_ppm.cc +++ /dev/null @@ -1,532 +0,0 @@ -// Copyright (C) 2009 EPITA Research and Development Laboratory (LRDE) -// -// This file is part of Olena. -// -// Olena is free software: you can redistribute it and/or modify it under -// the terms of the GNU General Public License as published by the Free -// Software Foundation, version 2 of the License. -// -// Olena is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU -// General Public License for more details. -// -// You should have received a copy of the GNU General Public License -// along with Olena. If not, see <http://www.gnu.org/licenses/>. -// -// As a special exception, you may use this file as part of a free -// software project without restriction. Specifically, if other files -// instantiate templates or use macros or inline functions from this -// file, or you compile this file and link it with other files to produce -// an executable, this file does not by itself cause the resulting -// executable to be covered by the GNU General Public License. This -// exception does not however invalidate any other reasons why the -// executable file might be covered by the GNU General Public License. - -#include <libgen.h> -#include <iostream> - -#include <mln/core/image/image2d.hh> -#include <mln/core/image/imorph/tr_image.hh> -#include <mln/core/alias/neighb2d.hh> - -#include <mln/labeling/colorize.hh> - -#include <mln/io/pbm/all.hh> -#include <mln/io/ppm/save.hh> - -#include <mln/math/min.hh> - -#include <mln/logical/not.hh> - -#include <mln/fun/v2v/rgb_to_int_u.hh> - -#include <mln/literal/colors.hh> -#include <mln/value/rgb8.hh> -#include <mln/value/label_16.hh> - -#include <mln/draw/box.hh> - -#include <mln/geom/translate.hh> - -#include <scribo/binarization/sauvola.hh> - -#include <scribo/draw/bounding_boxes.hh> - -#include <scribo/primitive/extract/objects.hh> - -#include <scribo/primitive/link/merge_double_link.hh> -#include <scribo/primitive/link/with_single_left_link.hh> -#include <scribo/primitive/link/with_single_right_link.hh> - -#include <scribo/primitive/group/apply.hh> -#include <scribo/primitive/group/from_double_link.hh> -#include <scribo/primitive/group/from_single_link.hh> - -#include <scribo/filter/objects_with_holes.hh> -#include <scribo/filter/object_links_bbox_h_ratio.hh> -#include <scribo/filter/object_links_bbox_overlap.hh> - -#include <scribo/filter/common/objects_photo.hh> - -#include <scribo/filter/object_groups_small.hh> -#include <scribo/filter/object_groups_v_thickness.hh> - -#include <scribo/debug/decision_image.hh> -#include <scribo/debug/save_bboxes_image.hh> -#include <scribo/debug/save_linked_bboxes_image.hh> - -#include <scribo/debug/usage.hh> - -#include <scribo/preprocessing/split_bg_fg.hh> - -#include <scribo/make/debug_filename.hh> - -#include <mln/util/timer.hh> -#include <mln/core/var.hh> - -const char *args_desc[][2] = -{ - { "input.ppm", "A color image." }, - { "debug_output_dir", "Directory were debug images will be saved" }, - { "lambda", "Lambda value used for foreground extraction" }, - {0, 0} -}; - - -namespace mln -{ - - struct mask_non_text : Function_v2v<mask_non_text> - { - typedef value::rgb8 result; - typedef image2d<bool> I; - - mask_non_text(const image2d<bool>& mask) - : mask_(mask), p_(mask_) - { - p_.start(); - } - - result operator()(const result& v) const - { - bool b = p_.val(); - p_.next(); - if (!b) - return v / 2; - else - return v; - - } - - I mask_; - mutable mln_pixter_(I) p_; - }; - - - template <typename I, typename L> - mln_concrete(I) - compute_highlight_image(const I& input_rgb, - const object_image<L>& objects) - { - mln_ch_value(I, bool) mask; - initialize(mask, input_rgb); - data::fill(mask, false); - - for_all_components(i, objects.bboxes()) - data::fill((mask | objects.bbox(i)).rw(), true); - - mask_non_text f(mask); - mln_concrete(I) output = data::transform(input_rgb, f); - - for_all_components(i, objects.bboxes()) - mln::draw::box(output, objects.bbox(i), literal::red); - - return output; - } - - - template <typename I, typename L> - mln_concrete(I) - compute_text_image(const I& input_rgb, - const object_image<L>& grouped_objects) - { - const util::array<mln_domain(L)>& bboxes = grouped_objects.bboxes(); - - unsigned shift = 5; - float height = 1, width = 0; - for_all_components(i, bboxes) - { - height += bboxes(i).nrows() + shift; - width = math::max(static_cast<float>(bboxes(i).ncols()), width); - } - if (width == 0) - width = 1; - - I output(height, width); - data::fill(output, literal::black); - - algebra::vec<2, float> dv; - dv[0] = 0; - dv[1] = 0; - for_all_ncomponents(i, grouped_objects.nlabels()) - { - mln_VAR(tmp, duplicate(input_rgb | grouped_objects.bbox(i))); - - typedef fun::x2x::translation<mln_site_(I)::dim, float> trans_t; - trans_t trans(dv - grouped_objects.bbox(i).pmin().to_vec()); - - mln_domain(I) tr_box(grouped_objects.bbox(i).pmin().to_vec() + trans.t(), - grouped_objects.bbox(i).pmax().to_vec() + trans.t()); - - tr_image<mln_domain(I), tmp_t, trans_t> tr_ima(tr_box, tmp, trans); - - data::paste(tr_ima, output); - dv[0] += grouped_objects.bbox(i).nrows() + shift; - } - - return output; - } - -} // end of namespace mln - - -int main(int argc, char* argv[]) -{ - using namespace scribo; - using namespace mln; - - if (argc != 3 && argc != 4 && argc != 5 && argc != 6) - return scribo::debug::usage(argv, - "Find text in a photo.", - "input.ppm output.ppm [bg/fg] [debug_output_dir] [lambda]", - args_desc, - "A color image where the text is highlighted."); - - if (argc > 4) - scribo::make::internal::debug_filename_prefix = argv[4]; - - trace::entering("main"); - - image2d<value::rgb8> input_rgb; - io::ppm::load(input_rgb, argv[1]); - - - unsigned lambda; - if (argc == 6) - lambda = atoi(argv[5]); - else - lambda = 1.2 * (input_rgb.nrows() + input_rgb.ncols()); - - - util::timer timer_; - - // Extract foreground - image2d<value::int_u8> intensity_ima; - std::cout << "Extracting foreground..." << std::endl; - timer_.start(); - - if (argc > 3 && atoi(argv[3]) != 0) - { - // Extract foreground - timer_.start(); - image2d<value::rgb8> - fg = preprocessing::split_bg_fg(input_rgb, - lambda, - 32).second(); - intensity_ima = data::transform(fg, mln::fun::v2v::rgb_to_int_u<8>()); - float t_ = timer_; - std::cout << "Foreground extracted. " << t_ << std::endl; - } - else - { - timer_.start(); - intensity_ima = data::transform(input_rgb, - mln::fun::v2v::rgb_to_int_u<8>()); - float t_ = timer_; - std::cout << "Intensity image " << t_ << std::endl; - } - float t_ = timer_; - std::cout << "Foreground extracted. " << t_ << std::endl; - - // Binarize foreground to use it in the processing chain. - // FIXME: TOO SLOW! - std::cout << "Binarizing foreground..." << std::endl; - timer_.restart(); - image2d<bool> input = binarization::sauvola(intensity_ima, 11); - io::pbm::save(input, "input.pbm"); - t_ = timer_; - std::cout << "Foreground binarized. " << t_ << std::endl; - - - - typedef image2d<value::label_16> L; - - /// Finding objects. - std::cout << "Extracting objects..." << std::endl; - timer_.restart(); - value::label_16 nobjects; - object_image(L) - objects = scribo::primitive::extract::objects(input, c8(), nobjects); - t_ = timer_; - std::cout << "Object extracted" << t_ << std::endl; - - /// First filtering. - std::cout << "Filtering objects..." << std::endl; - timer_.restart(); - object_image(L) filtered_objects = filter::common::objects_photo(objects); - t_ = timer_; - std::cout << "Object filtered" << t_ << std::endl; - - - - /// linking potential objects - std::cout << "Linking objects..." << std::endl; - timer_.restart(); - object_links<L> left_link - = primitive::link::with_single_left_link(filtered_objects, 30); - t_ = timer_; - std::cout << "Left Link done" << t_ << std::endl; - - timer_.restart(); - object_links<L> right_link - = primitive::link::with_single_right_link(filtered_objects, 30); - t_ = timer_; - std::cout << "Right Link done" << t_ << std::endl; - - - -#ifndef NOUT - if (argc > 3) - { - std::cerr << "BEFORE - nobjects = " << nobjects << std::endl; - scribo::debug::save_linked_bboxes_image(input, - filtered_objects, - left_link, right_link, - literal::red, literal::cyan, - literal::yellow, - literal::green, - scribo::make::debug_filename("links.ppm")); - } -#endif - - - - // Validating left and right links. - timer_.restart(); - object_links<L> - merged_links = primitive::link::merge_double_link(filtered_objects, - left_link, - right_link); - t_ = timer_; - std::cout << "Right/Left Validation. " << t_ << std::endl; - - - // Remove links if bboxes have too different sizes. - std::cout << "Filtering object links..." << std::endl; - timer_.restart(); - object_links<L> - hratio_filtered_links = filter::object_links_bbox_h_ratio(filtered_objects, - merged_links, - 1.50f); - - - - -#ifndef NOUT - if (argc > 3) - { - image2d<value::rgb8> - hratio_decision_image = scribo::debug::decision_image(input, - merged_links, - hratio_filtered_links); - io::ppm::save(hratio_decision_image, - scribo::make::debug_filename("hratio_links_decision_image.ppm")); - } -#endif - - - - - //Remove links if bboxes overlap too much. - object_links<L> overlap_filtered_links - = filter::object_links_bbox_overlap(filtered_objects, - hratio_filtered_links, - 0.80f); - - - - -#ifndef NOUT - if (argc > 3) - { - image2d<value::rgb8> overlap_decision_image - = scribo::debug::decision_image(input, - hratio_filtered_links, - overlap_filtered_links); - io::ppm::save(overlap_decision_image, - scribo::make::debug_filename("overlap_links_decision_image.ppm")); - } -#endif - - - t_ = timer_; - std::cout << "Objects links filtered. " << t_ << std::endl; - - std::cout << "Grouping objects..." << std::endl; - timer_.restart(); - object_groups<L> - groups = primitive::group::from_single_link(filtered_objects, - overlap_filtered_links); - - - - /// Apply grouping in a temporary image (for debug purpose). - object_image(L) - raw_group_image = primitive::group::apply(filtered_objects, - filter::object_groups_small(groups, 2)); - - t_ = timer_; - std::cout << "Objects grouped. " << t_ << std::endl; - -#ifndef NOUT - - if (argc > 3) - scribo::debug::save_bboxes_image(input, - raw_group_image.bboxes(), - literal::red, - scribo::make::debug_filename("group_image.ppm")); -#endif // !NOUT - - std::cout << "Filtering groups..." << std::endl; - - util::timer g_timer; - - timer_.restart(); - // Remove objects part of groups with strictly less than 3 objects. - - g_timer.start(); - object_groups<L> - filtered_small_groups = filter::object_groups_small(groups, 3); - t_ = g_timer; - std::cout << "Small groups removed " << t_ << std::endl; - - - -#ifndef NOUT - image2d<value::rgb8> decision_image; - if (argc > 3) - { - decision_image = scribo::debug::decision_image(input, - groups, - filtered_small_groups); - io::ppm::save(decision_image, - scribo::make::debug_filename("small_groups_filter.ppm")); - } -#endif - - - - // Remove objects part of groups having a mean thickness lower than 8. - g_timer.restart(); - object_groups<L> filtered_thin_groups - = filter::object_groups_v_thickness(filtered_small_groups, 8); - t_ = g_timer; - std::cout << "Groups too thin " << t_ << std::endl; - - -#ifndef NOUT - if (argc > 3) - { - decision_image = scribo::debug::decision_image(input, - groups, - filtered_thin_groups); - - io::ppm::save(decision_image, - scribo::make::debug_filename("thin_groups_filter.ppm")); - } -#endif - - - - /// Apply grouping in the object image. - g_timer.restart(); - - object_image(L) - grouped_objects = primitive::group::apply(filtered_objects, - filtered_thin_groups); - - t_ = g_timer; - std::cout << "Group applied to object image " << t_ << std::endl; - - /// Objects have been grouped. We try to link groups together. - /// This time a single link is enough since non-wanted objects have - /// been removed. - g_timer.restart(); - left_link - = primitive::link::with_single_left_link(grouped_objects, 30); - - /// Grouping groups. - groups = primitive::group::from_single_link(grouped_objects, left_link); - - - grouped_objects = primitive::group::apply(grouped_objects, groups); - - t_ = g_timer; - std::cout << "Link and group again " << t_ << std::endl; - - timer_.stop(); - - io::ppm::save(mln::labeling::colorize(value::rgb8(), - grouped_objects, - grouped_objects.nlabels()), - scribo::make::debug_filename("out_before_hole_filter.ppm")); - - - timer_.resume(); - - g_timer.restart(); - /// Filter grouped objects not having enough background components. - grouped_objects = scribo::filter::objects_with_holes_slow(grouped_objects, 2); -// grouped_objects = scribo::filter::objects_with_holes(grouped_objects, 2, 2); - t_ = g_timer; - std::cout << "Objects_with_holes " << t_ << std::endl; - - t_ = timer_; - std::cout << "Objects groups filtered. " << t_ << std::endl; - -#ifndef NOUT - if (argc > 3) - { - scribo::draw::bounding_boxes(decision_image, raw_group_image, literal::yellow); - scribo::draw::bounding_boxes(decision_image, grouped_objects, literal::blue); - io::ppm::save(decision_image, scribo::make::debug_filename("decision_image.ppm")); - - std::cerr << "AFTER - nobjects = " << grouped_objects.nlabels() << std::endl; - } -#endif - - - std::cout << "Saving result..." << std::endl; - io::ppm::save(mln::labeling::colorize(value::rgb8(), - grouped_objects, - grouped_objects.nlabels()), - argv[2]); - -#ifndef NOUT - io::ppm::save(compute_highlight_image(input_rgb, grouped_objects), - scribo::make::debug_filename("orig_with_bboxes.ppm")); -// scribo::debug::save_bboxes_image(input_rgb, grouped_objects.bboxes(), -// literal::red, -// scribo::make::debug_filename("orig_with_bboxes.ppm")); -#endif - - io::ppm::save(compute_text_image(input_rgb, grouped_objects), - scribo::make::debug_filename("out_text.ppm")); - - - std::cout << "# objects = " << grouped_objects.nlabels() << std::endl; - - trace::exiting("main"); - return grouped_objects.nlabels() != 0; -} diff --git a/scribo/src/text_in_photo_fast.cc b/scribo/src/text_in_picture.cc similarity index 99% rename from scribo/src/text_in_photo_fast.cc rename to scribo/src/text_in_picture.cc index d61cc6f..8ad4697 100644 --- a/scribo/src/text_in_photo_fast.cc +++ b/scribo/src/text_in_picture.cc @@ -106,6 +106,7 @@ // #include <scribo/core/line_set.hh> // #include <scribo/text/recognition.hh> +// #include <scribo/text/merging.hh> const char *args_desc[][2] = { @@ -697,6 +698,7 @@ Common usage: ./text_in_photo_fast input.ppm output.ppm 1 1 1 1 1", // scribo::line_set<L> // lines = scribo::make::line_set(groups); +// lines = scribo::text::merging(lines); // text::recognition(lines, "fra", "out.txt"); trace::exiting("main"); diff --git a/scribo/test.cc b/scribo/test.cc deleted file mode 100644 index 462d9b5..0000000 --- a/scribo/test.cc +++ /dev/null @@ -1,44 +0,0 @@ -// Copyright (C) 2009 - EPITA Research and Development Laboratory (LRDE) -// -// This file is part of Olena. -// -// Olena is free software: you can redistribute it and/or modify it under -// the terms of the GNU General Public License as published by the Free -// Software Foundation, version 2 of the License. -// -// Olena is distributed in the hope that it will be useful, -// but WITHOUT ANY WARRANTY; without even the implied warranty of -// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU -// General Public License for more details. -// -// You should have received a copy of the GNU General Public License -// along with Olena. If not, see <http://www.gnu.org/licenses/>. -// -// As a special exception, you may use this file as part of a free -// software project without restriction. Specifically, if other files -// instantiate templates or use macros or inline functions from this -// file, or you compile this file and link it with other files to produce -// an executable, this file does not by itself cause the resulting -// executable to be covered by the GNU General Public License. This -// exception does not however invalidate any other reasons why the -// executable file might be covered by the GNU General Public License. - -#include <mln/essential/2d.hh> -#include <mln/util/ord.hh> - -int main() -{ - using namespace mln; - - - point2d p1(0, 2); - point2d p2(0, 2); - - std::cout << util::ord_strict(p1, p2) << std::endl; - std::cout << util::ord_weak(p1, p2) << std::endl; - - std::cout << util::ord_strict(p2, p1) << std::endl; - std::cout << util::ord_weak(p2, p1) << std::endl; - -} -- 1.5.6.5