Mercurial > repos > ImageTools / file comparison
comparison: test/denoise.jl
test/denoise.jl
- changeset 25
- 1fba916f2b77
- parent 11
- f1bbdf68f35b
- child 28
- d1f40f6654cb
equal
deleted
inserted
replaced
| 24:f6630036e588 | 25:1fba916f2b77 |
|---|---|
| 3 ################## | 3 ################## |
| 4 | 4 |
| 5 __precompile__() | 5 __precompile__() |
| 6 | 6 |
| 7 using Printf | 7 using Printf |
| 8 using Images | 8 using FileIO |
| 9 using ColorTypes: Gray | |
| 10 # ColorVectorSpace is only needed to ensure that conversions | |
| 11 # between different ColorTypes are defined. | |
| 12 import ColorVectorSpace | |
| 9 import TestImages | 13 import TestImages |
| 10 | 14 |
| 11 using AlgTools.Util | 15 using AlgTools.Util |
| 12 using AlgTools.LinkedLists | 16 using AlgTools.LinkedLists |
| 13 using ImageTools.Denoise | 17 using ImageTools.Denoise |
| 23 accel = true, | 27 accel = true, |
| 24 noise_level = 0.5, | 28 noise_level = 0.5, |
| 25 verbose_iter = 10, | 29 verbose_iter = 10, |
| 26 maxiter = 1000, | 30 maxiter = 1000, |
| 27 save_results = false, | 31 save_results = false, |
| 28 save_iterations = false, | |
| 29 image_name = "lighthouse", | 32 image_name = "lighthouse", |
| 33 save_iterations = false | |
| 30 ) | 34 ) |
| 31 | 35 |
| 32 ####################### | 36 ####################### |
| 33 # Main testing routine | 37 # Main testing routine |
| 34 ####################### | 38 ####################### |
| 42 # Parameters for this experiment | 46 # Parameters for this experiment |
| 43 params = default_params ⬿ kwargs | 47 params = default_params ⬿ kwargs |
| 44 params = params ⬿ (save_prefix = save_prefix * params.image_name,) | 48 params = params ⬿ (save_prefix = save_prefix * params.image_name,) |
| 45 | 49 |
| 46 # Load image and add noise | 50 # Load image and add noise |
| 47 b = Float64.(Gray.(TestImages.testimage(params.image_name))) | 51 b = Float64.(Gray{Float64}.(TestImages.testimage(params.image_name))) |
| 48 b_noisy = b .+ params.noise_level.*randn(size(b)...) | 52 b_noisy = b .+ params.noise_level.*randn(size(b)...) |
| 49 | 53 |
| 50 # Launch (background) visualiser | 54 # Launch (background) visualiser |
| 51 st, iterate = initialise_visualisation(visualise) | 55 st, iterate = initialise_visualisation(visualise) |
| 52 | 56 |
| 55 | 59 |
| 56 if params.save_results | 60 if params.save_results |
| 57 perffile = params.save_prefix * ".txt" | 61 perffile = params.save_prefix * ".txt" |
| 58 println("Saving " * perffile) | 62 println("Saving " * perffile) |
| 59 write_log(perffile, st.log, "# params = $(params)\n") | 63 write_log(perffile, st.log, "# params = $(params)\n") |
| 64 fn = (t, ext) -> "$(params.save_prefix)_$(t).$(ext)" | |
| 65 save(File(format"PNG", fn("true", "png")), grayimg(b)) | |
| 66 save(File(format"PNG", fn("data", "png")), grayimg(b_noisy)) | |
| 67 save(File(format"PNG", fn("reco", "png")), grayimg(x)) | |
| 60 end | 68 end |
| 61 | 69 |
| 62 # Exit background visualiser | 70 # Exit background visualiser |
| 63 finalise_visualisation(st) | 71 finalise_visualisation(st) |
| 64 end | 72 end |
