--- milena/mln/binarization/tmms_hysteresis.hh | 740 ++++++++++++++++------------ 1 file changed, 413 insertions(+), 327 deletions(-) diff --git a/milena/mln/binarization/tmms_hysteresis.hh b/milena/mln/binarization/tmms_hysteresis.hh index f80caf5..36a7cf9 100644 --- a/milena/mln/binarization/tmms_hysteresis.hh +++ b/milena/mln/binarization/tmms_hysteresis.hh @@ -24,355 +24,441 @@ // executable file might be covered by the GNU General Public License. #ifndef MLN_BINARIZATION_TMMS_HYSTERESIS_HH -# define MLN_BINARIZATION_TMMS_HYSTERESIS_HH +#define MLN_BINARIZATION_TMMS_HYSTERESIS_HH /// \file /// /// \brief Binarization using TMMS algorithm combined with hysteresis. -# include <mln/core/concept/image.hh> -# include <mln/core/concept/window.hh> -# include <mln/core/alias/neighb2d.hh> -# include <mln/morpho/dilation.hh> -# include <mln/morpho/erosion.hh> -# include <mln/labeling/blobs.hh> +#include <mln/core/concept/image.hh> +#include <mln/core/concept/window.hh> +#include <mln/core/alias/neighb2d.hh> +#include <mln/morpho/dilation.hh> +#include <mln/morpho/erosion.hh> +#include <mln/labeling/blobs.hh> -namespace mln -{ +namespace mln { - namespace binarization - { + namespace binarization { - /*! - Binarize \p input by the use of TMMS algorithm (see "Text - Segmentation in Natural Scenes Using Toggle-Mapping" from - icip09 for more details \cite fabrizio2009icip). This version - is combined with hysteresis. + /*! + Binarize \p input by the use of TMMS algorithm (see "Text + Segmentation in Natural Scenes Using Toggle-Mapping" from + icip09 for more details \cite fabrizio2009icip). This version + is combined with hysteresis. - \param[in] input the input image. + \param[in] input the input image. - \param[in] win the structuring element used to compute erosion - and dilation of input image (During TMMS algorithm, input image - is mapped over two fuctions: an erosion and a dilation of the - input image). + \param[in] win the structuring element used to compute erosion + and dilation of input image (During TMMS algorithm, input image + is mapped over two fuctions: an erosion and a dilation of the + input image). - \param[in] c_min the min contrast allowed to extract feature. + \param[in] c_min the min contrast allowed to extract feature. - \return an image with "foreground" marked with the highest - possible color (hc), the "background" marked with the lowest - possible color (lc) and homogeneous regions of the image marked - with the color (hc+lc)/2. (This will fail with two colors - images, but it seems to be a bit stupid to try to binarize two - color images?) + \return an image with "foreground" marked with the highest + possible color (hc), the "background" marked with the lowest + possible color (lc) and homogeneous regions of the image marked + with the color (hc+lc)/2. (This will fail with two colors + images, but it seems to be a bit stupid to try to binarize two + color images?) - \author J. Fabrizio + \author J. Fabrizio - \ingroup modroutines - */ - template<typename I, typename W> - mln_concrete(I) - tmms_hysteresis(const Image<I>& input_, const Window<W>& win, - const mln_value(I)& c_min_low, - const mln_value(I)& c_min_high, - const mln_value(I)& min_level, - const mln_value(I)& med_level, - const mln_value(I)& max_level, - const int pourcent); - - -# ifndef MLN_INCLUDE_ONLY - - // Implementations - - namespace impl - { - - namespace generic - { + \ingroup modroutines + */ + template<typename I, typename W> + mln_concrete(I) + tmms_hysteresis(const Image<I>& input_, const Window<W>& win, + const mln_value(I) & c_min_low, + const mln_value(I) & c_min_high, + const mln_value(I) & min_level, + const mln_value(I) & med_level, + const mln_value(I) & max_level, + const int pourcent); + + +#ifndef MLN_INCLUDE_ONLY + + // Implementations + + namespace impl { + + namespace generic { + + template<typename I, typename W> + mln_concrete(I) + tmms_hysteresis(const Image<I>& input_, const Window<W>& win_, + const mln_value(I) & c_min_low, + const mln_value(I) & c_min_high, + const mln_value(I) & min_level, + const mln_value(I) & med_level, + const mln_value(I) & max_level, + const int pourcent) { + mln_trace("mln::binarization::impl::generic::tmms_hysteresis"); + + const I& input = exact(input_); + const W& win = exact(win_); + + mln_precondition(input.is_valid()); + mln_precondition(win.is_valid()); + + typedef mln_value(I) V; + + mln_concrete(I) dilation = mln::morpho::dilation(input, win); + mln_concrete(I) erosion = mln::morpho::erosion(input, win); + + mln_concrete(I) low_image; + initialize(low_image, input); + mln_concrete(I) high_image; + initialize(high_image, input); + + // mln_piter(I) p(input.domain()); + // + // for_all(p) { + // V dil = dilation(p); // by def : dil>=ima>=ero + // V ero = erosion(p); + // V ima = input(p); + // + // if ((V) (dil - ero) < c_min_low) { + // low_image(p) = med_level; + // high_image(p) = med_level; + // } else if ((V) (dil - ero) < c_min_high) { + // low_image(p) = med_level; + // if ((V) (dil - ima) < pourcent * ((V) (dil - ero)) / 100) + // high_image(p) = max_level; + // else + // high_image(p) = min_level; + // } else { + // if ((V) (dil - ima) < pourcent * ((V) (dil - ero)) / 100) { + // low_image(p) = max_level; + // high_image(p) = max_level; + // } else { + // low_image(p) = min_level; + // high_image(p) = min_level; + // } + // } + // + // } + // + // unsigned nlabels; + // mln_ch_value(I, unsigned) + // labels_high_images = mln::labeling::blobs(high_image, mln::c4(), nlabels); + // + // std::vector<V> values(nlabels + 1); + // + // for (unsigned i = 0; i <= nlabels; i++) + // values[i] = med_level; + // + // for_all(p) + // if (low_image(p) != med_level) + // values[labels_high_images(p)] = low_image(p); + // + // for_all(p) + // if (low_image(p) == med_level) + // low_image(p) = values[labels_high_images(p)]; + // + // return low_image; + + //OPTIM MYRIAM + + mln_pixter(I) px_in(input); + mln_pixter(I) px_dil(dilation); + mln_pixter(I) px_ero(erosion); + mln_pixter(I) px_low(low_image); + mln_pixter(I) px_hig(high_image); + + for_all_5(px_in, px_dil, px_ero, px_low, px_hig) { + + V dil = px_dil.val(); // by def : dil>=ima>=ero + V ero = px_ero.val(); + V ima = px_in.val(); + + if ((V) (dil - ero) < c_min_low) { + px_low.val() = med_level; + px_hig.val() = med_level; + } else if ((V) (dil - ero) < c_min_high) { + px_low.val() = med_level; + if ((V) (dil - ima) < pourcent * ((V) (dil - ero)) / 100) + px_hig.val() = max_level; + else + px_hig.val() = min_level; + } else { + if ((V) (dil - ima) < pourcent * ((V) (dil - ero)) / 100) { + px_low.val() = max_level; + px_hig.val() = max_level; + } else { + px_low.val() = min_level; + px_hig.val() = min_level; + } + } + } + + unsigned nlabels; + mln_ch_value(I, unsigned) + labels_high_images = mln::labeling::blobs(high_image, mln::c4(), nlabels); + mln_pixter(I) px_lab(labels_high_images); + + std::vector<V> values(nlabels + 1); + + for (unsigned i = 0; i <= nlabels; i++) + values[i] = med_level; + + for_all_2(px_low, px_lab) + if (px_low.val() != med_level) + values[px_lab.val()] = px_low.val(); + + for_all_2(px_low, px_lab) + if (px_low.val() == med_level) + px_low.val() = values[px_lab.val()]; + + return low_image; + } + + } // end of namespace mln::binarization::impl::generic + + template<typename I, typename W> + mln_concrete(I) + tmms_hysteresis_fastest(const Image<I>& input_, const Window<W>& win_, + const mln_value(I) & c_min_low, const mln_value(I) & c_min_high, + const mln_value(I) & min_level, + const mln_value(I) & med_level, + const mln_value(I) & max_level, + const int pourcent) { + mln_trace("mln::binarization::impl::tmms_hysteresis_fastest"); + + const I& input = exact(input_); + const W& win = exact(win_); + + mln_precondition(input.is_valid()); + mln_precondition(win.is_valid()); + + typedef mln_value(I) V; + + mln_concrete(I) dilation = mln::morpho::dilation(input, win); + mln_concrete(I) erosion = mln::morpho::erosion(input, win); + + // Checking border sizes. + if (dilation.border() != input.border()) + border::resize(dilation, input.border()); + if (erosion.border() != input.border()) + border::resize(erosion, input.border()); + mln_assertion(input.border() == dilation.border()); + mln_assertion(input.border() == erosion.border()); + + mln_concrete(I) low_image; + initialize(low_image, input); + mln_concrete(I) high_image; + initialize(high_image, input); + +// unsigned nrows = geom::nrows(input); +// unsigned ncols = geom::ncols(input); +// for (unsigned row = 0; row < nrows; ++row) { +// const V *p_input = &(input.at_(row, 0)); +// const V *p_dilation = &(dilation.at_(row, 0)); +// const V *p_erosion = &(erosion.at_(row, 0)); +// V *p_low_image = &(low_image.at_(row, 0)); +// V *p_high_image = &(high_image.at_(row, 0)); +// +// for (unsigned col = 0; col < ncols; ++col) { +// V dil = *p_dilation; // by def : dil>=ima>=ero +// V ero = *p_erosion; +// V ima = *p_input; +// +// if ((V) (dil - ero) < c_min_low) { +// *p_low_image = med_level; +// *p_high_image = med_level; +// } else if ((V) (dil - ero) < c_min_high) { +// *p_low_image = med_level; +// +// if ((V) (dil - ima) < (V) (pourcent * (dil - ero) / 100)) +// *p_high_image = max_level; +// else +// *p_high_image = min_level; +// } else { +// if ((V) (dil - ima) < (V) (pourcent * (dil - ero) / 100)) { +// *p_low_image = max_level; +// *p_high_image = max_level; +// } else { +// *p_low_image = min_level; +// *p_high_image = min_level; +// } +// } +// +// ++p_input; +// ++p_dilation; +// ++p_erosion; +// ++p_low_image; +// ++p_high_image; +// } +// } +// unsigned nlabels; +// mln_ch_value(I, unsigned) +// labels_high_images = mln::labeling::blobs(high_image, c4(), nlabels); + + // OPTIM MYRIAM + mln_pixter(I) px_low(low_image); + mln_pixter(I) px_hig(high_image); + + unsigned nrows = geom::nrows(input); + unsigned ncols = geom::ncols(input); + for (unsigned row = 0; row < nrows; ++row) { + const V *p_input = &(input.at_(row, 0)); + const V *p_dilation = &(dilation.at_(row, 0)); + const V *p_erosion = &(erosion.at_(row, 0)); + V *p_low_image = &(low_image.at_(row, 0)); + V *p_high_image = &(high_image.at_(row, 0)); + + for (unsigned col = 0; col < ncols; ++col) { + V dil = *p_dilation; // by def : dil>=ima>=ero + V ero = *p_erosion; + V ima = *p_input; + + if ((V) (dil - ero) < c_min_low) { + *p_low_image = med_level; + *p_high_image = med_level; + } else if ((V) (dil - ero) < c_min_high) { + *p_low_image = med_level; + + if ((V) (dil - ima) < (V) (pourcent * (dil - ero) / 100)) + *p_high_image = max_level; + else + *p_high_image = min_level; + } else { + if ((V) (dil - ima) < (V) (pourcent * (dil - ero) / 100)) { + *p_low_image = max_level; + *p_high_image = max_level; + } else { + *p_low_image = min_level; + *p_high_image = min_level; + } + } + + ++p_input; + ++p_dilation; + ++p_erosion; + ++p_low_image; + ++p_high_image; + } + } + + unsigned nlabels; + mln_ch_value(I, unsigned) + labels_high_images = mln::labeling::blobs(high_image, c4(), nlabels); + mln_pixter(image2d<unsigned int>) px_lab(labels_high_images); + + std::vector<V> values(nlabels + 1); + + for (unsigned i = 0; i <= nlabels; ++i) + values[i] = med_level; + +// mln_piter(I) p(input.domain()); +// for_all(p) +// if (low_image(p) != med_level) +// values[labels_high_images(p)] = low_image(p); +// +// for_all(p) +// if (low_image(p) == med_level) +// low_image(p) = values[labels_high_images(p)]; + + //OPTIM MYRIAM + + for_all_2(px_low, px_lab) + if (px_low.val() != med_level) + values[px_lab.val()] = px_low.val(); + + for_all_2(px_low, px_lab) + if (px_low.val() == med_level) + px_low.val() = values[px_lab.val()]; + + return low_image; + } + + }// end of namespace mln::binarization::impl + + + // Dispatch + + namespace internal { + + template<typename I, typename W> + mln_concrete(I) + tmms_hysteresis_dispatch(const I& ima, const W& win, + const mln_value(I) & c_min_low, + const mln_value(I) & c_min_high, + const mln_value(I) & min_level, + const mln_value(I) & med_level, + const mln_value(I) & max_level, + const int pourcent, mln::trait::image::speed::any) { + return impl::generic::tmms_hysteresis(ima, win, + c_min_low, c_min_high, + min_level, med_level, + max_level, pourcent); + } + + template<typename I, typename W> + mln_concrete(I) + tmms_hysteresis_dispatch(const I& ima, const W& win, + const mln_value(I) & c_min_low, + const mln_value(I) & c_min_high, + const mln_value(I) & min_level, + const mln_value(I) & med_level, + const mln_value(I) & max_level, + const int pourcent, mln::trait::image::speed::fastest) { + return impl::tmms_hysteresis_fastest(ima, win, + c_min_low, c_min_high, + min_level, med_level, + max_level, pourcent); + } + + template<typename I, typename W> + mln_concrete(I) + tmms_hysteresis_dispatch(const I& ima, const W& win, + const mln_value(I) & c_min_low, + const mln_value(I) & c_min_high, + const mln_value(I) & min_level, + const mln_value(I) & med_level, + const mln_value(I) & max_level, + const int pourcent) { + return tmms_hysteresis_dispatch(ima, win, + c_min_low, c_min_high, + min_level, med_level, + max_level, pourcent, + mln_trait_image_speed(I)()); + } + + } //end of namespace namespace mln::binarization::internal + + + // Facade template<typename I, typename W> mln_concrete(I) tmms_hysteresis(const Image<I>& input_, const Window<W>& win_, - const mln_value(I)& c_min_low, - const mln_value(I)& c_min_high, - const mln_value(I)& min_level, - const mln_value(I)& med_level, - const mln_value(I)& max_level, - const int pourcent) - { - mln_trace("mln::binarization::impl::generic::tmms_hysteresis"); - - const I& input = exact(input_); - const W& win = exact(win_); - - mln_precondition(input.is_valid()); - mln_precondition(win.is_valid()); - - typedef mln_value(I) V; - - mln_concrete(I) dilation = mln::morpho::dilation(input, win); - mln_concrete(I) erosion = mln::morpho::erosion(input, win); - - mln_concrete(I) low_image; - initialize(low_image, input); - mln_concrete(I) high_image; - initialize(high_image, input); - - mln_piter(I) p(input.domain()); - for_all(p) - { - V dil = dilation(p); // by def : dil>=ima>=ero - V ero = erosion(p); - V ima = input(p); - - if ((V)(dil - ero) < c_min_low) - { - low_image(p) = med_level; - high_image(p) = med_level; - } - else if ((V)(dil - ero) < c_min_high) - { - low_image(p) = med_level; - if ((V)(dil - ima) < pourcent * ((V) (dil - ero)) / 100) - high_image(p) = max_level; - else - high_image(p) = min_level; - } - else - { - if ((V)(dil - ima) < pourcent * ((V)(dil - ero)) / 100) - { - low_image(p) = max_level; - high_image(p) = max_level; - } - else - { - low_image(p) = min_level; - high_image(p) = min_level; - } - } - - } - - unsigned nlabels; - mln_ch_value(I, unsigned) - labels_high_images = mln::labeling::blobs(high_image, mln::c4(), nlabels); - - std::vector<V> values(nlabels + 1); - - for (unsigned i = 0; i <= nlabels ; i++) - values[i] = med_level; - - for_all(p) - if (low_image(p) != med_level) - values[labels_high_images(p)] = low_image(p); - - for_all(p) - if (low_image(p) == med_level) - low_image(p) = values[labels_high_images(p)]; - - return low_image; + const mln_value(I) & c_min_low, const mln_value(I) & c_min_high, + const mln_value(I) & min_level, + const mln_value(I) & med_level, + const mln_value(I) & max_level, + const int pourcent) { + mln_trace("mln::binarization::tmms_hysteresis"); + + const I& input = exact(input_); + const W& win = exact(win_); + + mln_precondition(input.is_valid()); + mln_precondition(win.is_valid()); + + return internal::tmms_hysteresis_dispatch(input, win, + c_min_low, c_min_high, + min_level, med_level, + max_level, pourcent); } - } // end of namespace mln::binarization::impl::generic - - - template<typename I, typename W> - mln_concrete(I) - tmms_hysteresis_fastest(const Image<I>& input_, const Window<W>& win_, - const mln_value(I)& c_min_low, const mln_value(I)& c_min_high, - const mln_value(I)& min_level, - const mln_value(I)& med_level, - const mln_value(I)& max_level, - const int pourcent) - { - mln_trace("mln::binarization::impl::tmms_hysteresis_fastest"); - - const I& input = exact(input_); - const W& win = exact(win_); - - mln_precondition(input.is_valid()); - mln_precondition(win.is_valid()); - - typedef mln_value(I) V; - - mln_concrete(I) dilation = mln::morpho::dilation(input, win); - mln_concrete(I) erosion = mln::morpho::erosion(input, win); - - // Checking border sizes. - border::resize(dilation, input.border()); - border::resize(erosion, input.border()); - mln_assertion(input.border() == dilation.border()); - mln_assertion(input.border() == erosion.border()); - - mln_concrete(I) low_image; - initialize(low_image, input); - mln_concrete(I) high_image; - initialize(high_image, input); - - unsigned nrows = geom::nrows(input); - unsigned ncols = geom::ncols(input); - for (unsigned row = 0; row < nrows; ++row) - { - const V *p_input = &(input.at_(row, 0)); - const V *p_dilation = &(dilation.at_(row, 0)); - const V *p_erosion = &(erosion.at_(row, 0)); - V *p_low_image = &(low_image.at_(row, 0)); - V *p_high_image = &(high_image.at_(row, 0)); - - for (unsigned col = 0; col < ncols; ++col) - { - V dil = *p_dilation; // by def : dil>=ima>=ero - V ero = *p_erosion; - V ima = *p_input; - - if ((V)(dil - ero) < c_min_low) - { - *p_low_image = med_level; - *p_high_image = med_level; - } - else if ((V)(dil - ero) < c_min_high) - { - *p_low_image = med_level; - - if ((V)(dil - ima) < (V)(pourcent * (dil - ero) / 100)) - *p_high_image = max_level; - else - *p_high_image = min_level; - } - else - { - if ((V)(dil - ima) < (V)(pourcent * (dil - ero) / 100)) - { - *p_low_image = max_level; - *p_high_image = max_level; - } - else - { - *p_low_image = min_level; - *p_high_image = min_level; - } - } - - ++p_input; - ++p_dilation; - ++p_erosion; - ++p_low_image; - ++p_high_image; - } - } - - unsigned nlabels; - mln_ch_value(I, unsigned) - labels_high_images = mln::labeling::blobs(high_image, c4(), nlabels); - - std::vector<V> values(nlabels + 1); - - for(unsigned i = 0; i <= nlabels; ++i) - values[i] = med_level; - - mln_piter(I) p(input.domain()); - for_all(p) - if (low_image(p) != med_level) - values[labels_high_images(p)] = low_image(p); - - for_all(p) - if (low_image(p) == med_level) - low_image(p) = values[labels_high_images(p)]; - - return low_image; - } - - }// end of namespace mln::binarization::impl - - - // Dispatch - - namespace internal - { - - template<typename I, typename W> - mln_concrete(I) - tmms_hysteresis_dispatch(const I& ima, const W& win, - const mln_value(I)& c_min_low, - const mln_value(I)& c_min_high, - const mln_value(I)& min_level, - const mln_value(I)& med_level, - const mln_value(I)& max_level, - const int pourcent, mln::trait::image::speed::any) - { - return impl::generic::tmms_hysteresis(ima, win, - c_min_low, c_min_high, - min_level, med_level, - max_level, pourcent); - } - - template<typename I, typename W> - mln_concrete(I) - tmms_hysteresis_dispatch(const I& ima, const W& win, - const mln_value(I)& c_min_low, - const mln_value(I)& c_min_high, - const mln_value(I)& min_level, - const mln_value(I)& med_level, - const mln_value(I)& max_level, - const int pourcent, mln::trait::image::speed::fastest) - { - return impl::tmms_hysteresis_fastest(ima, win, - c_min_low, c_min_high, - min_level, med_level, - max_level, pourcent); - } - - template<typename I, typename W> - mln_concrete(I) - tmms_hysteresis_dispatch(const I& ima, const W& win, - const mln_value(I)& c_min_low, - const mln_value(I)& c_min_high, - const mln_value(I)& min_level, - const mln_value(I)& med_level, - const mln_value(I)& max_level, - const int pourcent) - { - return tmms_hysteresis_dispatch(ima, win, - c_min_low, c_min_high, - min_level, med_level, - max_level, pourcent, - mln_trait_image_speed(I)()); - } - - } //end of namespace namespace mln::binarization::internal - - - // Facade - - template<typename I, typename W> - mln_concrete(I) - tmms_hysteresis(const Image<I>& input_, const Window<W>& win_, - const mln_value(I)& c_min_low, const mln_value(I)& c_min_high, - const mln_value(I)& min_level, - const mln_value(I)& med_level, - const mln_value(I)& max_level, - const int pourcent) - { - mln_trace("mln::binarization::tmms_hysteresis"); - - const I& input = exact(input_); - const W& win = exact(win_); - - mln_precondition(input.is_valid()); - mln_precondition(win.is_valid()); - - return internal::tmms_hysteresis_dispatch(input, win, - c_min_low, c_min_high, - min_level, med_level, - max_level, pourcent); - } - -# endif // ! MLN_INCLUDE_ONLY - - } // end of namespace mln::binarization +#endif // ! MLN_INCLUDE_ONLY + + } // end of namespace mln::binarization } // end of namespace mln -- 1.7.10.4