* lazzara/igr.cc: new file. --- milena/sandbox/ChangeLog | 6 ++ milena/sandbox/lazzara/igr.cc | 106 +++++++++++++++++++++++++++++++++++++++++ 2 files changed, 112 insertions(+), 0 deletions(-) create mode 100644 milena/sandbox/lazzara/igr.cc diff --git a/milena/sandbox/ChangeLog b/milena/sandbox/ChangeLog index 7e2cd40..0d25be8 100644 --- a/milena/sandbox/ChangeLog +++ b/milena/sandbox/ChangeLog @@ -1,3 +1,9 @@ +2009-01-27 Guillaume Lazzara <z@lrde.epita.fr> + + First draft of a program for IGR's images. + + * lazzara/igr.cc: new file. + 2009-01-27 Thierry Geraud <thierry.geraud@lrde.epita.fr> Update color segmentation. diff --git a/milena/sandbox/lazzara/igr.cc b/milena/sandbox/lazzara/igr.cc new file mode 100644 index 0000000..2579fa6 --- /dev/null +++ b/milena/sandbox/lazzara/igr.cc @@ -0,0 +1,106 @@ +#include <mln/essential/2d.hh> +#include <mln/binarization/binarization.hh> +#include <mln/labeling/background.hh> + +#include <mln/debug/colorize.hh> + +#include <mln/registration/registration.hh> + +struct threshold : mln::Function_p2b<threshold> +{ + bool operator()(const mln::value::rgb8& val) const + { + unsigned inty = (val.red() + val.blue() + val.green()); + return inty > 100 && inty < 600; + } +}; + + + + +template <typename I> +mln_concrete(I) +fill_holes(const mln::Image<I>& ima) +{ + using namespace mln; + + mln_concrete(I) output; + initialize(output, ima); + data::fill(output, true); + + value::label_16 nlabels; + mln_ch_value(I, value::label_16) lbl = labeling::background(ima, c4(), nlabels); + + mln_VAR(lbl_count, labeling::compute(accu::meta::count(), ima, lbl, nlabels)); + + unsigned max_id = 1; + for (unsigned i = 2; i <= nlabels; ++i) + if (lbl_count[max_id] < lbl_count[i]) + max_id = i; + + data::fill((output | pw::value(lbl) == max_id).rw(), false); + + return output; +} + + +template <typename I> +mln_concrete(I) +keep_largest_component(const mln::Image<I>& ima) +{ + using namespace mln; + + image2d<bool> in_bw_cleaned = fill_holes(ima); + io::pbm::save(in_bw_cleaned, "0x_in_bw_cleaned.pbm"); + + logical::not_inplace(in_bw_cleaned); + image2d<bool> in_bw_cleaned_full = fill_holes(in_bw_cleaned); + io::pbm::save(in_bw_cleaned_full, "0x_in_bw_cleaned_full.pbm"); + + logical::not_inplace(in_bw_cleaned_full); + return in_bw_cleaned_full; +} + +int main(int argc, char* argv[]) +{ + using namespace mln; + using value::rgb8; + using value::int_u8; + using value::label_16; + + //Load image + typedef image2d<rgb8> I; + typedef image2d<bool> J; + I in; + io::ppm::load(in, argv[1]); + + threshold f; + J in_bw = binarization::binarization(in, f); + io::pbm::save(in_bw, "01_in_bw.pbm"); + + J ima = keep_largest_component(in_bw); + io::pbm::save(in_bw, "02_ima.pbm"); + + std::cout << "Compute gradient" << std::endl; + J ima_grad = morpho::gradient(ima, win_c4p()); + io::pbm::save(ima_grad, "03_ima_grad.pbm"); + + I ref; + io::ppm::load(ref, argv[2]); + J ref_bw = binarization::binarization(ref, f); + std::cout << "Computing registration..." << std::endl; + J cloud = ima_grad | pw::value(ima_grad) == true; + mln_VAR(registration, registration::registration(cloud, ref_bw)); + mln_VAR(icp, registration::icp(convert::to_p_array(cloud), ref_bw)); + + std::cout << "Build result image" << std::endl; + J res; + initialize(res, ima_grad); + mln_piter_(J) p(ima_grad.domain()); + for_all(p) + if (res.has(registration(p.to_site().to_vec()))) + res(registration(p.to_site().to_vec())) = ima_grad(p); + + io::pbm::save(res, "04_registered.pbm"); + +} -- 1.5.6.5