8 Mar
2013
8 Mar
'13
8:10 p.m.
---
scribo/sandbox/icdar_13_table/Makefile | 18 +++
scribo/sandbox/icdar_13_table/src/main.cc | 171 +++++++++++++++++++++++++++++
2 files changed, 189 insertions(+), 0 deletions(-)
create mode 100644 scribo/sandbox/icdar_13_table/Makefile
create mode 100644 scribo/sandbox/icdar_13_table/src/main.cc
diff --git a/scribo/sandbox/icdar_13_table/Makefile
b/scribo/sandbox/icdar_13_table/Makefile
new file mode 100644
index 0000000..2420eec
--- /dev/null
+++ b/scribo/sandbox/icdar_13_table/Makefile
@@ -0,0 +1,18 @@
+CCACHE=
+CC=g++
+CFLAGS=-Wall -Werror -O3
+CLIBS=-I../../../milena/ -I../../
+CLEAN=*.o $(OUTPUT) output/* log final.xml
+
+SRC=src/main.cc
+OUTPUT=table
+
+all: table
+
+table:
+ $(CCACHE) $(CC) $(CFLAGS) -DHAVE_TESSERACT_3 $(CLIBS) $(SRC) -ltesseract -o $(OUTPUT)
+
+clean:
+ rm -rf $(CLEAN)
+
+.PHONY: table
diff --git a/scribo/sandbox/icdar_13_table/src/main.cc
b/scribo/sandbox/icdar_13_table/src/main.cc
new file mode 100644
index 0000000..2ed125d
--- /dev/null
+++ b/scribo/sandbox/icdar_13_table/src/main.cc
@@ -0,0 +1,171 @@
+#include <mln/binarization/all.hh>
+
+#include <mln/core/image/image2d.hh>
+#include <mln/data/all.hh>
+
+#include <mln/fun/v2v/rgb_to_luma.hh>
+
+#include <mln/io/pbm/all.hh>
+#include <mln/io/ppm/all.hh>
+
+#include <mln/labeling/all.hh>
+#include <mln/logical/and.hh>
+
+#include <mln/value/all.hh>
+
+#include <tesseract/baseapi.h>
+
+#include <scribo/binarization/sauvola.hh>
+#include <scribo/core/component_set.hh>
+#include <scribo/preprocessing/denoise_fg.hh>
+#include <scribo/primitive/extract/all.hh>
+
+using namespace mln;
+
+void start_xml(std::ofstream& xml, const char* name, const char* pdf)
+{
+ xml.open(name);
+ xml << "<?xml version\"1.0\"
encoding=\"UTF-8\"?>" << std::endl
+ << "<document filename='" << pdf <<
"'>" << std::endl;
+}
+
+void end_xml(std::ofstream& xml)
+{
+ xml << "</document>" << std::endl;
+ xml.close();
+}
+
+void write_table(std::ofstream& xml, const point2d& start, const point2d&
end)
+{
+ static unsigned table = 0;
+ static unsigned region = 0;
+ static unsigned page = 1;
+
+ xml << "\t<table id='" << table <<
"'>" << std::endl
+ << "\t\t<region id='" << region << "'
page='" << page << "'>" << std::endl
+ << "\t\t<bounding-box x1='" << start[1] <<
"' y1='" << start[0] << "' "
+ << "x2='" << end[1] << "' y2='"
<< end[0] << "'/>" << std::endl
+ << "\t\t</region>" << std::endl;
+
+ ++table;
+}
+
+template<typename L>
+void get_horizontal_lines(scribo::component_set<L>& hlines,
+ image2d<bool>& ima,
+ image2d<bool>& bin_without,
+ const unsigned av_height)
+{
+ for (unsigned i = 1; i <= hlines.nelements(); ++i)
+ {
+ const box2d& b = hlines(i).bbox();
+
+ data::fill((bin_without | b).rw(), false);
+
+ if (b.height() != 0 && b.width() != 0 && b.height() < av_height)
+ data::fill((ima | b).rw(), true);
+ }
+}
+
+template<typename L>
+void get_vertical_lines(scribo::component_set<L>& vlines,
+ image2d<bool>& ima,
+ image2d<bool>& bin_without,
+ const unsigned av_width,
+ const int delta_prox_h)
+{
+ for (unsigned i = 1; i <= vlines.nelements(); ++i)
+ {
+ const box2d& b = vlines(i).bbox();
+
+ data::fill((bin_without | b).rw(), false);
+
+ if (b.height() != 0 && b.width() != 0
+ && b.width() < av_width)
+ if (vlines(i).mass_center()[1] - vlines(i - 1).mass_center()[1] > delta_prox_h
+ || vlines(i).mass_center()[1] < vlines(i - 1).mass_center()[1])
+ data::fill((ima | b).rw(), true);
+ }
+}
+
+int main(int argc, char** argv)
+{
+ typedef value::label_16 V;
+ typedef image2d<V> L;
+
+ image2d<value::rgb8> original;
+ image2d<value::int_u8> filtered;
+ image2d<bool> bin, ima_hlines, ima_vlines, ima_tables, ima_texts,
+ bin_without_lines, bin_without_lines_denoised,
+ mask;
+ image2d<unsigned> labeled;
+ scribo::component_set< image2d<unsigned> > masks;
+ V nhlines, nvlines;
+ L hlines_ima, vlines_ima;
+ scribo::component_set<L> hlines, vlines;
+
+ // PARAMETERS
+ unsigned av_height = 4;
+ unsigned av_width = 4;
+ int delta_prox_h = 5;
+ unsigned min_height = 71;
+ unsigned min_width = 31;
+
+ // Loadin and binarization
+ std::ofstream xml;
+ start_xml(xml, "final.xml", argv[1]);
+
+ io::ppm::load(original, argv[1]);
+ filtered = data::transform(original,
mln::fun::v2v::rgb_to_luma<value::int_u8>());
+
+ bin = scribo::binarization::sauvola(filtered, 81, 0.44);
+
+ initialize(mask, bin);
+ initialize(ima_texts, bin);
+ initialize(ima_tables, bin);
+ data::fill(ima_tables, false);
+
+ bin_without_lines = duplicate(bin);
+
+ // Lines extraction
+ hlines = scribo::primitive::extract::lines_h_discontinued(bin, c4(), nhlines,
min_width, 2);
+ vlines = scribo::primitive::extract::lines_v_discontinued(bin, c4(), nvlines,
min_height, 2);
+
+ get_horizontal_lines(hlines, ima_tables, bin_without_lines, av_height);
+ get_vertical_lines(vlines, ima_tables, bin_without_lines, av_width, delta_prox_h);
+
+ // Denoising
+ bin_without_lines_denoised = scribo::preprocessing::denoise_fg(bin_without_lines, c8(),
3);
+
+ // Set the tables mask
+ unsigned n;
+ labeled = labeling::blobs(ima_tables, c8(), n);
+ masks = scribo::component_set< image2d<unsigned> >(labeled, n);
+ data::fill(mask, false);
+
+ for (unsigned i = 1; i <= masks.nelements(); ++i)
+ data::fill((mask | masks(i).bbox()).rw(), true);
+
+ // Compose table zones with bin_without_lines_lines
+ ima_texts = logical::and_(bin_without_lines_denoised, mask);
+
+ // Get lines images
+ hlines_ima = hlines.labeled_image();
+ vlines_ima = vlines.labeled_image();
+ ima_hlines = data::convert(bool(), hlines_ima);
+ ima_vlines = data::convert(bool(), vlines_ima);
+
+ // Write images and close XML
+ io::pbm::save(bin, "output/0_bin.pbm");
+ io::pbm::save(bin_without_lines, "output/1_bin_without_lines.pbm");
+ io::pbm::save(bin_without_lines_denoised,
"output/2_bin_without_lines_denoised.pbm");
+ io::pbm::save(ima_hlines, "output/3_hlines.pbm");
+ io::pbm::save(ima_vlines, "output/4_vlines.pbm");
+ io::pbm::save(ima_tables, "output/5_tables.pbm");
+ io::pbm::save(mask, "output/6_mask.pbm");
+ io::pbm::save(ima_texts, "output/7_texts.pbm");
+
+ end_xml(xml);
+
+ return 0;
+}
--
1.7.2.5