How to create the affinity?

[sciai-lab]

How to create the affinity?

[constantinpape]

We use shift kernels to compute the affinities from ground-truth, see https://github.com/inferno-pytorch/neurofire/blob/master/neurofire/transform/segmentation.py#L110.

[celinezou]

Hi! Thanks for the quick reply. I am sorry but I met another problem when abstracting groud truth affinity. I found you used "train_loader = get_isbi_loader_3d(data_config)" in your code to get the training data and its label(ground truth affinity), so I used the same code and converted this torch dataloader to NumPy to get the ground truth affinity. But the data configuration template file has the following configurations. In this case, the size of ground truth affinity is (17, 12, 459, 459) not (17, 30, 512, 512) , which is the size of the raw data and predicted affinity you provide.

How can I generate ground truth affinity in size(17, 30, 512, 512)? I have tried to set the window size as (30, 512, 512), but the ground truth affinity I got from this setting has an invalid value on the boundary, like this (on the top):

It will be great if you can explain what does every parameter means in slicing_config and how should I set these parameters to get ground truth affinity in size(17, 30, 512, 512) like yours?

I really appreciate your time and help. Thank you so much!

Best

[constantinpape]

The affinities at the borders of the volume are invalid. That's why we use the padding.
Why do you need the affinities for the full volume? Training can only be done on smaller patches anyways.

In the slicing_config the parameters mean the following: window_size: size of the patch to be extracted. stride: the stride used to iterate the patch through the whole volume in order to extract patches. data_slice: the subvolume used for extracting patches in numpy slice syntax. Here, the whole volume is used. padding: padding applied to the volume, in np.pad syntax.

[celinezou]

Thank you so so so much! It works in my case now. One final question: is there a GPU version of mutex watershed?

[constantinpape]

Thank you so so so much! It works in my case now.

Great!

One final question: is there a GPU version of mutex watershed?

No, the mutex watershed is based on building a spanning forest using a version of kruskal's algorithm that can express repulsive interactions. This is a sequential algorithm and not suited for GPU computing.

[celinezou]

Thank you so much! 👍