Apply mask prior to segmentation? about connected-components-3d

[william-silversmith]

Perhaps that was more verbose than you needed, but yes, I think you understand the problem correctly now and this is normal behavior.

[william-silversmith]

You can try masking the source image with numpy operators before applying cc3d. e.g. img *= boolean_mask You may also find useful masking operators in this other library I wrote called fastremap. Let me know if that helps!

[william-silversmith]

The algorithm skips regions that are colored black (0), so masking will speed up the run time.

[atpolonsky]

I'm still getting a returned feature of size equal to the number of black (0 value) voxels in my volume, even after applying a mask. Is this the expected behavior?

[william-silversmith]

Can you show me some of your code? How are you measuring the feature and performing the mask? Have you visualized the volume?

[atpolonsky]

I think I figured it out, the labels coming out of cc3d.connected_components has a region 0, so what I was doing subsequently was counting this as a feature (np.unique includes 0 as a unique number in the array). I can't actually tell if the algorithm is checking black voxels in this case, but I did see a slight slowdown when I made all of the voxels non-zero and ran the code again, maybe a second or two difference in run time at most. My test dataset volume was 1500 x 500 x 800 voxels, but I have larger volumes on the order of 5000x5000x500 voxels, so its possible without masking these would take much longer. If this behavior makes sense you can mark this as closed.

[william-silversmith]

cc3d takes a multi-labeled image as input and counts black voxels in that image as background. Non-zero voxels are counted as foreground, though neighboring values that do not match are not counted as part of the same component. It outputs an image with the foreground voxels labeled 1 to N where there are N connected regions of foreground voxels. Background labels (black voxels) remain unchanged and are treated specially so that the innermost loop skips over them without applying the decision tree. Therefore, you should expect better performance on an image that has a lot of black voxels.

connected-components-3d/cc3d.hpp

Lines 307 to 309 in 1b7479c

	if (cur == 0) {
	continue;
	}

Since the output relabeled image retains the black voxels, np.uniquewill indeed output an accounting of them. You can try using fastremap.unique for a faster implementation of that function.

https://github.com/seung-lab/fastremap/blob/master/fastremap.pyx#L705

As a 2D example:

INPUT IMAGE                OUTPUT IMAGE

0 0 4 5 6 4                0 0 1 2 3 4
0 4 0 5 6 4                0 1 0 2 3 4
0 4 0 5 6 4  ====cc3d===>  0 1 0 2 3 4
0 0 0 5 5 6                0 0 0 2 2 3
0 0 0 6 6 0                0 0 0 3 3 0

Applying np.unique(labels, return_counts=True) to the output image in the above example would return results agnostic to the connected components problem and would contain a tabulation of black pixels.