Olena-patches

olena-patches@lrde.epita.fr

9625 discussions

olena: olena-2.0-568-gc9c37e9 Build the set of detected columns during table recognition.

by Roland Levillain

* scribo/sandbox/icdar_13_table/src/new.cc: Here. * scribo/sandbox/icdar_13_table/src/disjoint_set.hh: New. --- scribo/sandbox/icdar_13_table/src/disjoint_set.hh | 54 +++++++++++++++++++++ scribo/sandbox/icdar_13_table/src/new.cc | 36 ++++++++++++++ 2 files changed, 90 insertions(+), 0 deletions(-) create mode 100644 scribo/sandbox/icdar_13_table/src/disjoint_set.hh diff --git a/scribo/sandbox/icdar_13_table/src/disjoint_set.hh b/scribo/sandbox/icdar_13_table/src/disjoint_set.hh new file mode 100644 index 0000000..b8ae409 --- /dev/null +++ b/scribo/sandbox/icdar_13_table/src/disjoint_set.hh @@ -0,0 +1,54 @@ +#ifndef DISJOINT_SET_HH +# define DISJOINT_SET_HH + +/// \file +/// Implementation of a disjoint-set forest. + +# include <limits> +# include <vector> + +template <typename T> +struct disjoint_set +{ + disjoint_set(size_t size) + // The maximum value of T is considered an ``invalid value''. + : parent(size, std::numeric_limits<T>::max()) + { + for (unsigned i = 0; i < parent.size(); ++i) + make_set(i); + } + + /// Create initial sets (singletons). + void make_set(T x) + { + parent[x] = x; + } + + bool is_root(T x) + { + return parent[x] == x; + } + + /// Find the root of the set containing \a x. + T find(T x) + { + if (!is_root(x)) + // Path compression. + parent[x] = find(parent[x]); + return parent[x]; + } + + /// Merge the set containing \a y into the set containing \a x. + void make_union(T x, T y) + { + T x_root = find(x); + T y_root = find(y); + parent[y_root] = x_root; + } + + /// Parent relationship. A node with is its own parent is a + /// root (representative of a component). + std::vector<T> parent; +}; + +#endif // ! DISJOINT_SET_HH diff --git a/scribo/sandbox/icdar_13_table/src/new.cc b/scribo/sandbox/icdar_13_table/src/new.cc index 2c41e31..bcf127e 100644 --- a/scribo/sandbox/icdar_13_table/src/new.cc +++ b/scribo/sandbox/icdar_13_table/src/new.cc @@ -54,6 +54,8 @@ #include <scribo/text/extract_paragraphs.hh> #include <scribo/text/merging.hh> +#include "disjoint_set.hh" + using namespace mln; // Draw weighted boxes (red < orange < cyan < green) @@ -558,6 +560,39 @@ int main(int argc, char** argv) } } + // Build the columns. + // First pass. + disjoint_set<node_id> columns_set(groups.nelements()); + for (unsigned i = 0; i < groups.nelements(); ++i) + { + const group_set& successors = nodes_below[i]; + for (group_set::const_iterator j = successors.begin(); + j != successors.end(); ++j) + columns_set.make_union(i, *j); + } + // Second pass: assign labels. Label 0 is unused and means + // ``default''. + std::vector<group_set> descendants(groups.nelements()); + for (unsigned i = 0; i < groups.nelements(); ++i) + { + // Process groups in reverse order. + unsigned j = groups.nelements() - 1 - i; + if (!columns_set.is_root(j)) + descendants[columns_set.find(j)].insert(j); + } + + // Visualization: columns. + image2d<value::rgb8> ima_columns = data::convert(value::rgb8(), bin_merged); + for (unsigned i = 0; i < groups.nelements(); ++i) + if (!descendants[i].empty()) + { + box2d column_box = groups(i).bbox(); + for (group_set::const_iterator j = descendants[i].begin(); + j != descendants[i].end(); ++j) + column_box.merge(groups(*j).bbox()); + draw::box(ima_columns, column_box, literal::red); + } + // Write images and close XML std::ostringstream path; unsigned number = 0; @@ -572,6 +607,7 @@ int main(int argc, char** argv) write_image(ima_links, "components", page, number, path); write_image(ima_groups, "groups", page, number, path); write_image(ima_valid, "valid", page, number, path); + write_image(ima_columns, "columns", page, number, path); } delete xml; -- 1.7.2.5

11 years, 7 months

olena: olena-2.0-569-g4ecdb6a Remove many false positives (intersecting boxes)

by Anthony Seure

--- scribo/sandbox/icdar_13_table/src/new.cc | 35 +++++++++++++++++++++++++++++- 1 files changed, 34 insertions(+), 1 deletions(-) diff --git a/scribo/sandbox/icdar_13_table/src/new.cc b/scribo/sandbox/icdar_13_table/src/new.cc index bcf127e..d52ac15 100644 --- a/scribo/sandbox/icdar_13_table/src/new.cc +++ b/scribo/sandbox/icdar_13_table/src/new.cc @@ -58,6 +58,13 @@ using namespace mln; +// Check if two boxes intersected +bool intersect(const box2d& b1, const box2d& b2) +{ + return ((unsigned) abs(b1.pcenter().col() - b2.pcenter().col()) * 2 < b1.width() + b2.width() + && (unsigned) abs(b1.pcenter().row() - b2.pcenter().row()) * 2 < b1.height() + b2.height()); +} + // Draw weighted boxes (red < orange < cyan < green) // 1 link < 2 links < 3 links < 3+ links template<typename T, typename L> @@ -581,6 +588,9 @@ int main(int argc, char** argv) descendants[columns_set.find(j)].insert(j); } + std::vector<box2d> columns; + std::set<unsigned> not_valid; + // Visualization: columns. image2d<value::rgb8> ima_columns = data::convert(value::rgb8(), bin_merged); for (unsigned i = 0; i < groups.nelements(); ++i) @@ -590,9 +600,32 @@ int main(int argc, char** argv) for (group_set::const_iterator j = descendants[i].begin(); j != descendants[i].end(); ++j) column_box.merge(groups(*j).bbox()); - draw::box(ima_columns, column_box, literal::red); + draw::box(ima_columns, column_box, literal::green); + columns.push_back(column_box); } + for (unsigned i = 0; i < columns.size(); ++i) + { + for (unsigned j = 0; j < columns.size(); ++j) + { + if (i != j) + { + const box2d& c1 = columns[i]; + const box2d& c2 = columns[j]; + + if (intersect(c1, c2) || intersect(c2, c1)) + { + not_valid.insert(i); + not_valid.insert(j); + } + } + } + } + + for (unsigned i = 0; i < columns.size(); ++i) + if (not_valid.find(i) != not_valid.end()) + draw::box(ima_columns, columns[i], literal::red); + // Write images and close XML std::ostringstream path; unsigned number = 0; -- 1.7.2.5

11 years, 7 months

olena: olena-2.0-570-g6836fdc Init scale space working directory

by Anthony Seure

--- milena/sandbox/anthony/Makefile | 21 +++++++++++++++++ milena/sandbox/anthony/src/scale_space.cc | 36 +++++++++++++++++++++++++++++ 2 files changed, 57 insertions(+), 0 deletions(-) create mode 100644 milena/sandbox/anthony/Makefile create mode 100644 milena/sandbox/anthony/src/scale_space.cc diff --git a/milena/sandbox/anthony/Makefile b/milena/sandbox/anthony/Makefile new file mode 100644 index 0000000..7f990da --- /dev/null +++ b/milena/sandbox/anthony/Makefile @@ -0,0 +1,21 @@ +CCACHE=ccache +CC=g++ +CFLAGS=-Wall -Werror -O3 -DNDEBUG +CLIBS=-I../../../scribo/ -I../../ +CLEAN=*.o output/* log final.xml *.pgm + +SRC=src/scale_space.cc +OUTPUT=a.out + +all: scale + +scale: + $(CCACHE) $(CC) $(CFLAGS) $(CLIBS) $(SRC) -o $(OUTPUT) + +clean: + rm -rf $(CLEAN) + +mrproper: clean + rm -f $(OUTPUT) + +.PHONY: scale clean mrproper diff --git a/milena/sandbox/anthony/src/scale_space.cc b/milena/sandbox/anthony/src/scale_space.cc new file mode 100644 index 0000000..9e452a3 --- /dev/null +++ b/milena/sandbox/anthony/src/scale_space.cc @@ -0,0 +1,36 @@ +#include <mln/binarization/all.hh> + +#include <mln/core/image/image2d.hh> + +#include <mln/data/all.hh> +#include <mln/draw/line.hh> + +#include <mln/fun/v2v/rgb_to_luma.hh> + +#include <mln/io/pgm/all.hh> + +#include <mln/labeling/all.hh> +#include <mln/literal/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; + +int main(int argc, char** argv) +{ + image2d<value::int_u8> original; + + io::pgm::load(original, "images/keith.pgm"); + + io::pgm::save(original, "final.pgm"); + + return 0; +} -- 1.7.2.5

11 years, 7 months

olena: olena-2.0-571-g929c08c Create blured copies of original image

by Anthony Seure

--- milena/sandbox/anthony/Makefile | 11 ++- milena/sandbox/anthony/src/scale_space.cc | 114 ++++++++++++++++++++++++----- 2 files changed, 103 insertions(+), 22 deletions(-) diff --git a/milena/sandbox/anthony/Makefile b/milena/sandbox/anthony/Makefile index 7f990da..81f788e 100644 --- a/milena/sandbox/anthony/Makefile +++ b/milena/sandbox/anthony/Makefile @@ -1,8 +1,8 @@ CCACHE=ccache CC=g++ -CFLAGS=-Wall -Werror -O3 -DNDEBUG -CLIBS=-I../../../scribo/ -I../../ -CLEAN=*.o output/* log final.xml *.pgm +CFLAGS=-Wall -Werror +CLIBS=-I../../ +CLEAN=*.o output/* log SRC=src/scale_space.cc OUTPUT=a.out @@ -10,7 +10,10 @@ OUTPUT=a.out all: scale scale: - $(CCACHE) $(CC) $(CFLAGS) $(CLIBS) $(SRC) -o $(OUTPUT) + $(CCACHE) $(CC) $(CFLAGS) -O2 -DNDEBUG $(CLIBS) $(SRC) -o $(OUTPUT) + +debug: + $(CCACHE) $(CC) $(CFLAGS) -g -ggdb $(CLIBS) $(SRC) -o $(OUTPUT) clean: rm -rf $(CLEAN) diff --git a/milena/sandbox/anthony/src/scale_space.cc b/milena/sandbox/anthony/src/scale_space.cc index 9e452a3..31180af 100644 --- a/milena/sandbox/anthony/src/scale_space.cc +++ b/milena/sandbox/anthony/src/scale_space.cc @@ -1,36 +1,114 @@ -#include <mln/binarization/all.hh> - #include <mln/core/image/image2d.hh> - +#include <mln/core/alias/neighb2d.hh> #include <mln/data/all.hh> -#include <mln/draw/line.hh> +#include <mln/io/pgm/all.hh> +#include <mln/literal/all.hh> +#include <mln/value/all.hh> +#include <mln/make/w_window2d.hh> +#include <mln/core/alias/w_window2d_int.hh> -#include <mln/fun/v2v/rgb_to_luma.hh> +#include <cmath> -#include <mln/io/pgm/all.hh> +using namespace mln; -#include <mln/labeling/all.hh> -#include <mln/literal/all.hh> -#include <mln/logical/and.hh> +// Set the gaussian kernel of size (6 * t + 1) +double *gaussian_kernel(unsigned t) +{ + int size = 6 * t + 1; + unsigned index = 0; + double *kernel = new double[size * size]; + static const double pi = 4.0 * atan(1.0); + double div = (2.0 * pi * t * t); -#include <mln/value/all.hh> + for (int i = -(size / 2); i < (size / 2) + 1; ++i) + { + for (int j = -(size / 2); j < (size / 2) + 1; ++j) + { + double e = exp(- (i * i + j * j) / (2.0 * t * t) ); + kernel[index++] = e / div; + } + } -#include <tesseract/baseapi.h> + return kernel; +} -#include <scribo/binarization/sauvola.hh> -#include <scribo/core/component_set.hh> -#include <scribo/preprocessing/denoise_fg.hh> -#include <scribo/primitive/extract/all.hh> +// Apply the convolution of the image by the kernel +template<typename I> +void convolve(const I& original, I& filtered, double *kernel, unsigned t) +{ + mln_piter(I) p(filtered.domain()); + int size = 6 * t + 1; + window2d win; -using namespace mln; + for (int i = -(size / 2); i < (size / 2) + 1; ++i) + { + for (int j = -(size / 2); j < (size / 2) + 1; ++j) + { + win.insert(i, j); + } + } + + + // Iterate through all image sites + for_all(p) + { + // Create the window around the site + mln_qiter(window2d) q(win, p); + float sum = 0; + int index = 0; + // Iterate through all window image sites + for_all(q) + { + if (filtered.has(q)) + sum += original(q) * kernel[index++]; + else + sum += 127 * kernel[index++]; + } + + filtered(p) = static_cast<int>(sum); + } +} + +// Blur the image thanks to a convolution matrix +template<typename I> +image2d<value::int_u8> blur(const I& original, unsigned t) +{ + I filtered; + double *kernel = gaussian_kernel(t); + + initialize(filtered, original); + convolve(original, filtered, kernel, t); + + return filtered; +} + +template<typename I> +void downscaleResolution(const I& original, I& toReduce) +{ +} + +// MAIN ENTRY POINT int main(int argc, char** argv) { image2d<value::int_u8> original; + io::pgm::load(original, "images/flower.pgm"); + + image2d<value::int_u8> blur1, blur2, blur4; + initialize(blur1, original); + initialize(blur2, original); + initialize(blur4, original); + + io::pgm::save(original, "output/original.pgm"); + + blur1 = blur(original, 1); + io::pgm::save(blur1, "output/blur1.pgm"); - io::pgm::load(original, "images/keith.pgm"); + blur2 = blur(original, 2); + io::pgm::save(blur2, "output/blur2.pgm"); - io::pgm::save(original, "final.pgm"); + blur4 = blur(original, 4); + io::pgm::save(blur4, "output/blur4.pgm"); return 0; } -- 1.7.2.5

11 years, 7 months

olena: olena-2.0-572-gcd3eaf5 Create a full scale space

by Anthony Seure

--- milena/sandbox/anthony/src/scale_space.cc | 100 +++++++++++++++++++++++------ 1 files changed, 80 insertions(+), 20 deletions(-) diff --git a/milena/sandbox/anthony/src/scale_space.cc b/milena/sandbox/anthony/src/scale_space.cc index 31180af..d7b0570 100644 --- a/milena/sandbox/anthony/src/scale_space.cc +++ b/milena/sandbox/anthony/src/scale_space.cc @@ -1,11 +1,6 @@ #include <mln/core/image/image2d.hh> -#include <mln/core/alias/neighb2d.hh> #include <mln/data/all.hh> #include <mln/io/pgm/all.hh> -#include <mln/literal/all.hh> -#include <mln/value/all.hh> -#include <mln/make/w_window2d.hh> -#include <mln/core/alias/w_window2d_int.hh> #include <cmath> @@ -72,7 +67,7 @@ void convolve(const I& original, I& filtered, double *kernel, unsigned t) // Blur the image thanks to a convolution matrix template<typename I> -image2d<value::int_u8> blur(const I& original, unsigned t) +I blur(const I& original, unsigned t) { I filtered; double *kernel = gaussian_kernel(t); @@ -83,32 +78,97 @@ image2d<value::int_u8> blur(const I& original, unsigned t) return filtered; } +// Downscale by 2 the image resolution template<typename I> -void downscaleResolution(const I& original, I& toReduce) +I downscaleResolution(const I& original) { + mln_piter(I) p(original.domain()); + window2d win; + int index = 0; + + // Initialize the rescaled image + box2d size(original.nrows() / 2, original.ncols() / 2); + I reduced(size); + + // Set the 2x2 window matrix + for (int i = 0; i < 2; ++i) + for (int j = 0; j < 2; ++j) + win.insert(i, j); + + // Iterate through all image sites + for_all(p) + { + // Get 1/4 of total pixels + if (index % 2 == 0) + { + int i = (index % original.ncols()) / 2; + int j = (index / original.ncols()) / 2; + int count = 0; + int sum = 0; + mln_qiter(window2d) q(win, p); + + // Iterate through all window image sites + for_all(q) + { + if (original.has(q)) + { + sum += original(q); + ++count; + } + } + + if (reduced.has(point2d(j, i))) + opt::at(reduced, j, i) = sum / count; + } + + ++index; + } + + return reduced; } -// MAIN ENTRY POINT +// Build the n-th octave on b blur levels +template<typename I> +void buildOctave(const I& original, unsigned n, unsigned b) +{ + std::stringstream name; + I blured; + initialize(blured, original); + + name << "output/o" << n << "b0.pgm"; + io::pgm::save(original, name.str()); + + for(unsigned i = 1; i <= b; i *= 2) + { + blured = blur(original, i); + name.str(""); + name << "output/o" << n << "b" < original; io::pgm::load(original, "images/flower.pgm"); + const unsigned blur_level = 4; + unsigned octave = 1; - image2d<value::int_u8> blur1, blur2, blur4; - initialize(blur1, original); - initialize(blur2, original); - initialize(blur4, original); - - io::pgm::save(original, "output/original.pgm"); + // 1st octave + buildOctave(original, octave++, blur_level); - blur1 = blur(original, 1); - io::pgm::save(blur1, "output/blur1.pgm"); + // 2nd octave + image2d<value::int_u8> reduced = downscaleResolution(original); + buildOctave(reduced, octave++, blur_level); - blur2 = blur(original, 2); - io::pgm::save(blur2, "output/blur2.pgm"); + // 3rd octave + reduced = downscaleResolution(reduced); + buildOctave(reduced, octave++, blur_level); - blur4 = blur(original, 4); - io::pgm::save(blur4, "output/blur4.pgm"); + // 4th octave + reduced = downscaleResolution(reduced); + buildOctave(reduced, octave++, blur_level); return 0; } -- 1.7.2.5

11 years, 7 months

olena: olena-2.0-586-g8255583 Fix merge conflicts with ICDAR13 recognition table work from Roland

by Anthony Seure

This is a merge commit message of the following branches: Merge: cd3eaf5 f16ceed hooks/post-receive -- Olena, a generic and efficient image processing platform

11 years, 7 months

olena: branch sandbox/anthony updated: olena-2.0-590-g8020a9b

by git＠lrde.epita.fr

This is an automated email from the git hooks/post-receive script. It was generated because a ref change was pushed to the repository containing the project "Olena, a generic and efficient image processing platform". The branch sandbox/anthony has been updated via 8020a9b93b5ccdafd6fcf810cc3cddafe70a1dae (commit) via 6cc7082289672e0ffce59cf96ff6e8eab5b9b98e (commit) via d0ee5384b5af5eb5a9df1d2121ac733d8c1bb0c4 (commit) via bacc2e1ef82b965c4f5c4455ce81c082ab26cbd3 (commit) from 82555836eee6be994349379073645c7ffb95efd2 (commit) Those revisions listed above that are new to this repository have not appeared on any other notification email; so we list those revisions in full, below. - Log ----------------------------------------------------------------- 8020a9b Fix "too many keypoints" error 6cc7082 Compute and display extremum BUT resolution scaling factor TO CHANGE (2 -> 1,5) d0ee538 Find extremum of the image bacc2e1 Build scale space inside a vector for use in extrema localization ----------------------------------------------------------------------- Summary of changes: milena/sandbox/anthony/Makefile | 6 +- milena/sandbox/anthony/src/scale_space.cc | 268 ++++++++++++++++++++++++----- 2 files changed, 230 insertions(+), 44 deletions(-) hooks/post-receive -- Olena, a generic and efficient image processing platform

11 years, 7 months

olena: olena-2.0-587-gbacc2e1 Build scale space inside a vector for use in extrema localization

by Anthony Seure

--- milena/sandbox/anthony/Makefile | 2 +- milena/sandbox/anthony/src/scale_space.cc | 60 ++++++++++++++++++++--------- 2 files changed, 43 insertions(+), 19 deletions(-) diff --git a/milena/sandbox/anthony/Makefile b/milena/sandbox/anthony/Makefile index 81f788e..558bef5 100644 --- a/milena/sandbox/anthony/Makefile +++ b/milena/sandbox/anthony/Makefile @@ -10,7 +10,7 @@ OUTPUT=a.out all: scale scale: - $(CCACHE) $(CC) $(CFLAGS) -O2 -DNDEBUG $(CLIBS) $(SRC) -o $(OUTPUT) + $(CCACHE) $(CC) $(CFLAGS) -O3 -DNDEBUG $(CLIBS) $(SRC) -o $(OUTPUT) debug: $(CCACHE) $(CC) $(CFLAGS) -g -ggdb $(CLIBS) $(SRC) -o $(OUTPUT) diff --git a/milena/sandbox/anthony/src/scale_space.cc b/milena/sandbox/anthony/src/scale_space.cc index d7b0570..61308c4 100644 --- a/milena/sandbox/anthony/src/scale_space.cc +++ b/milena/sandbox/anthony/src/scale_space.cc @@ -129,8 +129,12 @@ I downscaleResolution(const I& original) // Build the n-th octave on b blur levels template<typename I> -void buildOctave(const I& original, unsigned n, unsigned b) +void buildOctave(const I& original, + unsigned n, + unsigned b, + std::vector< std::vector >& scaleSpace) { + std::vector octave; std::stringstream name; I blured; initialize(blured, original); @@ -138,37 +142,57 @@ void buildOctave(const I& original, unsigned n, unsigned b) name << "output/o" << n << "b0.pgm"; io::pgm::save(original, name.str()); + octave.push_back(original); + for(unsigned i = 1; i <= b; i *= 2) { blured = blur(original, i); + octave.push_back(blured); + name.str(""); name << "output/o" << n << "b" < +void buildScaleSpace(std::vector< std::vector >& scaleSpace, + const I& original, + unsigned octave_level, + unsigned blur_level) { - image2d<value::int_u8> original; - io::pgm::load(original, "images/flower.pgm"); - const unsigned blur_level = 4; - unsigned octave = 1; + if (octave_level > 2) + { + // 1st octave + buildOctave(original, 1, blur_level, scaleSpace); - // 1st octave - buildOctave(original, octave++, blur_level); + // 2nd octave + I reduced = downscaleResolution(original); + buildOctave(reduced, 2, blur_level, scaleSpace); - // 2nd octave - image2d<value::int_u8> reduced = downscaleResolution(original); - buildOctave(reduced, octave++, blur_level); + for (unsigned i = 3; i <= octave_level; ++i) + { + // i-th octave + reduced = downscaleResolution(reduced); + buildOctave(reduced, i, blur_level, scaleSpace); + } + } +} + +// Main entry point +int main(int argc, char** argv) +{ + typedef image2d<value::int_u8> I; + const unsigned blur_level = 3; + const unsigned octave_level = 4; + std::vector< std::vector > scaleSpace; + I original; - // 3rd octave - reduced = downscaleResolution(reduced); - buildOctave(reduced, octave++, blur_level); + io::pgm::load(original, "images/flower.pgm"); - // 4th octave - reduced = downscaleResolution(reduced); - buildOctave(reduced, octave++, blur_level); + buildScaleSpace(scaleSpace, original, octave_level, blur_level); return 0; } -- 1.7.2.5

11 years, 7 months

olena: olena-2.0-588-gd0ee538 Find extremum of the image

by Anthony Seure

--- milena/sandbox/anthony/Makefile | 4 +- milena/sandbox/anthony/src/scale_space.cc | 171 ++++++++++++++++++++++++++--- 2 files changed, 157 insertions(+), 18 deletions(-) diff --git a/milena/sandbox/anthony/Makefile b/milena/sandbox/anthony/Makefile index 558bef5..e43db7f 100644 --- a/milena/sandbox/anthony/Makefile +++ b/milena/sandbox/anthony/Makefile @@ -9,10 +9,10 @@ OUTPUT=a.out all: scale -scale: +scale: clean $(CCACHE) $(CC) $(CFLAGS) -O3 -DNDEBUG $(CLIBS) $(SRC) -o $(OUTPUT) -debug: +debug: mrproper $(CCACHE) $(CC) $(CFLAGS) -g -ggdb $(CLIBS) $(SRC) -o $(OUTPUT) clean: diff --git a/milena/sandbox/anthony/src/scale_space.cc b/milena/sandbox/anthony/src/scale_space.cc index 61308c4..0eca2b6 100644 --- a/milena/sandbox/anthony/src/scale_space.cc +++ b/milena/sandbox/anthony/src/scale_space.cc @@ -1,6 +1,9 @@ #include <mln/core/image/image2d.hh> #include <mln/data/all.hh> #include <mln/io/pgm/all.hh> +#include <mln/io/ppm/save.hh> +#include <mln/core/alias/neighb2d.hh> +#include <mln/literal/all.hh> #include <cmath> @@ -36,13 +39,8 @@ void convolve(const I& original, I& filtered, double *kernel, unsigned t) window2d win; for (int i = -(size / 2); i < (size / 2) + 1; ++i) - { for (int j = -(size / 2); j < (size / 2) + 1; ++j) - { win.insert(i, j); - } - } - // Iterate through all image sites for_all(p) @@ -139,7 +137,7 @@ void buildOctave(const I& original, I blured; initialize(blured, original); - name << "output/o" << n << "b0.pgm"; + name << "output/SS-o" << n << "-0.pgm"; io::pgm::save(original, name.str()); octave.push_back(original); @@ -150,49 +148,190 @@ void buildOctave(const I& original, octave.push_back(blured); name.str(""); - name << "output/o" << n << "b" < void buildScaleSpace(std::vector< std::vector >& scaleSpace, const I& original, - unsigned octave_level, - unsigned blur_level) + unsigned n, + unsigned b) { - if (octave_level > 2) + if (n > 2) { // 1st octave - buildOctave(original, 1, blur_level, scaleSpace); + buildOctave(original, 1, b, scaleSpace); // 2nd octave I reduced = downscaleResolution(original); - buildOctave(reduced, 2, blur_level, scaleSpace); + buildOctave(reduced, 2, b, scaleSpace); - for (unsigned i = 3; i <= octave_level; ++i) + for (unsigned i = 3; i <= n; ++i) { // i-th octave reduced = downscaleResolution(reduced); - buildOctave(reduced, i, blur_level, scaleSpace); + buildOctave(reduced, i, b, scaleSpace); + } + } + else + { + std::cerr << "ERROR: buildScaleSpace: Wrong level number (" << n << ")" + << std::endl; + } +} + +// Build the DoG pyramid of the scale space +template<typename I> +void buildDifferenceOfGaussianSpace(std::vector< std::vector >& scaleSpace, + std::vector< std::vector >& dogSpace) +{ + for (unsigned i = 0; i < scaleSpace.size(); ++i) + { + std::vector octave = scaleSpace.at(i); + std::vector dogOctave; + + for (unsigned j = 0; j < octave.size() - 1; ++j) + { + std::stringstream name; + I level1 = octave.at(j); + I level2 = octave.at(j + 1); + I dog; + + initialize(dog, level1); + + mln_piter(I) p(level1.domain()); + + for_all(p) + dog(p) = level1(p) - level2(p); + + dogOctave.push_back(dog); + name << "output/DoG_o" < +void find_extremum(T& min, T& max, const I& current, N n) +{ + for_all(n) + { + if (current(n) > max) + max = current(n); + + if (current(n) < min) + min = current(n); } } +// Find the minimum value in the upper and lower layers around p +template<typename T, typename I, typename P> +void find_min(T& min, const I& upper, const I& lower, const P& p) +{ + mln_niter(neighb2d) n(c8(), p); + + for_all(n) + { + if (lower(n) < min) + min = lower(n); + + if (upper(n) < min) + min = upper(n); + } +} + +// Find the maximum value in the upper and lower layers around p +template<typename T, typename I, typename P> +void find_max(T& max, const I& upper, const I& lower, const P& p) +{ + mln_niter(neighb2d) n(c8(), p); + + for_all(n) + { + if (lower(n) > max) + max = lower(n); + + if (upper(n) > max) + max = upper(n); + } +} + +// Build the extrema image +template<typename I, typename C> +void buildExtrema(C extrema, + I original, + std::vector< std::vector >& dogSpace) +{ + extrema = data::convert(value::rgb8(), original); + value::int_u8 min = 255; + value::int_u8 max = 0; + value::int_u8 center = 0; + + for (unsigned i = 0; i < dogSpace.size(); ++i) + { + std::vector octave = dogSpace.at(i); + + for (unsigned j = 1; j < octave.size() - 1; ++j) + { + I current = octave.at(j); + I upper = octave.at(j+1); + I lower = octave.at(j-1); + + mln_piter(I) p(current.domain()); + + for_all(p) + { + mln_niter(neighb2d) n(c8(), p); + center = current(p); + + find_extremum(min, max, current, n); + + if (center <= min) + { + find_min(min, upper, lower, p); + + if (center <= min) + extrema(p) = literal::blue; + } + else if (center >= max) + { + find_max(max, upper, lower, p); + + if (center >= max) + extrema(p) = literal::red; + } + } + } + } + io::ppm::save(extrema, "output/extrema.ppm"); +} + // Main entry point int main(int argc, char** argv) { typedef image2d<value::int_u8> I; - const unsigned blur_level = 3; + typedef image2d<value::rgb8> C; + + const unsigned blur_level = 4; const unsigned octave_level = 4; std::vector< std::vector > scaleSpace; + std::vector< std::vector > dogSpace; I original; + C extrema; - io::pgm::load(original, "images/flower.pgm"); + io::pgm::load(original, "images/keith512.pgm"); buildScaleSpace(scaleSpace, original, octave_level, blur_level); + buildDifferenceOfGaussianSpace(scaleSpace, dogSpace); + buildExtrema(extrema, original, dogSpace); return 0; } -- 1.7.2.5

11 years, 7 months

olena: olena-2.0-589-g6cc7082 Compute and display extremum BUT resolution scaling factor TO CHANGE (2 -> 1, 5)

by Anthony Seure

--- milena/sandbox/anthony/src/scale_space.cc | 90 ++++++++++++++++++----------- 1 files changed, 56 insertions(+), 34 deletions(-) diff --git a/milena/sandbox/anthony/src/scale_space.cc b/milena/sandbox/anthony/src/scale_space.cc index 0eca2b6..074eff4 100644 --- a/milena/sandbox/anthony/src/scale_space.cc +++ b/milena/sandbox/anthony/src/scale_space.cc @@ -4,6 +4,7 @@ #include <mln/io/ppm/save.hh> #include <mln/core/alias/neighb2d.hh> #include <mln/literal/all.hh> +#include <mln/value/int_s8.hh> #include <cmath> @@ -207,7 +208,10 @@ void buildDifferenceOfGaussianSpace(std::vector< std::vector >& scaleSpace, mln_piter(I) p(level1.domain()); for_all(p) - dog(p) = level1(p) - level2(p); + { + int diff = ((int) level2(p)) - ((int) level1(p)); + dog(p) = (diff >= 0) ? diff : 0; + } dogOctave.push_back(dog); name << "output/DoG_o" << i << "-" << j << ".pgm"; @@ -234,34 +238,40 @@ void find_extremum(T& min, T& max, const I& current, N n) // Find the minimum value in the upper and lower layers around p template<typename T, typename I, typename P> -void find_min(T& min, const I& upper, const I& lower, const P& p) +//void find_min(T& min, const I& upper, const I& lower, const P& p) +void find_min(T& min, const I& upper, const I& lower, const P& pUpper, const P& pLower) { - mln_niter(neighb2d) n(c8(), p); - - for_all(n) - { - if (lower(n) < min) - min = lower(n); - - if (upper(n) < min) - min = upper(n); - } + mln_niter(neighb2d) nUpper(c8(), pUpper); + mln_niter(neighb2d) nLower(c8(), pLower); + + // Upper processing + for_all(nUpper) + if (upper(nUpper) < min) + min = upper(nUpper); + + // Lower processing + for_all(nLower) + if (lower(nLower) < min) + min = lower(nLower); } // Find the maximum value in the upper and lower layers around p template<typename T, typename I, typename P> -void find_max(T& max, const I& upper, const I& lower, const P& p) +//void find_max(T& max, const I& upper, const I& lower, const P& p) +void find_max(T& max, const I& upper, const I& lower, const P& pUpper, const P& pLower) { - mln_niter(neighb2d) n(c8(), p); - - for_all(n) - { - if (lower(n) > max) - max = lower(n); - - if (upper(n) > max) - max = upper(n); - } + mln_niter(neighb2d) nUpper(c8(), pUpper); + mln_niter(neighb2d) nLower(c8(), pLower); + + // Upper processing + for_all(nUpper) + if (upper(nUpper) > max) + max = upper(nUpper); + + // Lower processing + for_all(nLower) + if (lower(nLower) > max) + max = lower(nLower); } // Build the extrema image @@ -271,19 +281,21 @@ void buildExtrema(C extrema, std::vector< std::vector >& dogSpace) { extrema = data::convert(value::rgb8(), original); + //initialize(extrema, original); value::int_u8 min = 255; value::int_u8 max = 0; value::int_u8 center = 0; - for (unsigned i = 0; i < dogSpace.size(); ++i) - { - std::vector octave = dogSpace.at(i); + const unsigned blur_level = dogSpace.at(0).size(); //TODO Reverse the vector - for (unsigned j = 1; j < octave.size() - 1; ++j) + //unsigned i = 0; + for (unsigned i = 0; i < blur_level; ++i) + { + for (unsigned j = 1; j < dogSpace.size() - 1; ++j) { - I current = octave.at(j); - I upper = octave.at(j+1); - I lower = octave.at(j-1); + I current = dogSpace.at(j).at(i); + I upper = dogSpace.at(j+1).at(i); + I lower = dogSpace.at(j-1).at(i); mln_piter(I) p(current.domain()); @@ -292,25 +304,35 @@ void buildExtrema(C extrema, mln_niter(neighb2d) n(c8(), p); center = current(p); + mln_psite(I) pUpper(p.row() / 2, p.col() / 2); + mln_psite(I) pLower(p.row() * 2, p.col() * 2); + find_extremum(min, max, current, n); if (center <= min) { - find_min(min, upper, lower, p); + find_min(min, upper, lower, pUpper, pLower); if (center <= min) - extrema(p) = literal::blue; + { + mln_psite(I) pOriginal(p.row() * (j+1), p.col() * (j+1)); + extrema(pOriginal) = literal::blue; + } } else if (center >= max) { - find_max(max, upper, lower, p); + find_max(min, upper, lower, pUpper, pLower); if (center >= max) - extrema(p) = literal::red; + { + mln_psite(I) pOriginal(p.row() * (j+1), p.col() * (j+1)); + extrema(pOriginal) = literal::red; + } } } } } + io::ppm::save(extrema, "output/extrema.ppm"); } -- 1.7.2.5

11 years, 7 months

Jump to page:

2025

2024

2023

2022

2021

2020

2019

2018

2017

2016

2015

2014

2013

2012

2011

2010

2009

2008

2007

2006

2005

2004

Olena-patches