Learning Watershed Cuts Energy Functions

摘要

In recent work, several popular segmentation methods have been unified as energy minimization on a graph. In other work, supervised learning methods have been generalized from predicting labels to predicting structured, graph-like objects. A recent contribution to this second area showed how the Rand Index could be directly minimized when using Connected Components as a segmentation method. We build on this work and present an efficient mini-batch learning method for Connected Component segmentation and also show how it can be generalized to the Watershed Cuts segmentation method. We present initial results applying these new contributions to image segmentation problems in materials microscopy and discuss challenges and future directions.