fepegar / staple

Python implementation of the STAPLE segmentation algorithm
MIT License
24 stars 5 forks source link

Not getting the correct result #49

Closed JosephKKim closed 1 year ago

JosephKKim commented 1 year ago

Description

I am trying to make STAPLE ground truth using the source code you have provided. Dataset that I am considering is retinal vessel dataset (2D binary segmentation dataset) are listed below, which have 2 and 3 annotation respectively. ChaseDB1 STARE

What I Did

staple STARE_ah.png STARE_vk.png STARE_result.png STARE_staple.png

ValueError: Maximum number of iterations reached before convergence

I have converted ppm file into png file and ran staple with the command I have stated above.

And even when it converged it provided an image filled with black. Image_staple

In this case what would be the problem?

Thanks in advance!

fepegar commented 1 year ago

Hi, @JosephKKim. I won't be able to debug this. I suggest you use a different implementation, as suggested in the README. Good luck!