shameempk / fast_seg Goto Github PK

Everytime I run the program with python fast_seg.py -i lenna.png there is a prompt and I mark the bg and fg and then press esc. After that it throws the following error everytime irrespecctive of how i mark. I have also tried it with other images.

 File "fast_seg.py", line 389, in <module>
    main()
  File "fast_seg.py", line 337, in main
    G = gen_graph(I_lab, SP_list, hist_ob, hist_bg)
  File "fast_seg.py", line 230, in gen_graph
    pr_bg = int(hist_bg[l_i, a_i, b_i]) / hist_bg_sum
ZeroDivisionError: division by zero

Cold you plese help me out as I am also conducting research in this topic and woud love to use this code and cite this. Let me know if I'm doing something wrong or maybe there's some mistake in the algorithm.

Hello, thanks of your contribution about such good work.
I am using graph neural network to parcel brain region, however the result is not good, I want to use graph cut as post process.
The inputs of my current model are: adjacency matrix (10242, 10242), feature matrix (10242, 6), label matrix (10242). The output is the probability y that each node belongs to a label, and its dimension is (10242, 36)
I want to use graph cut to update y for better performance.

I have a few questions about your code:
First, most of graph cut is only for two categories. Can you do multi label tasks?
Second, if I want to do post-processing, what should my input be?