Train Image Segmentation Models to Accept User Feedback via Voronoi Tiling, Part 2

How to train an off-the-shelf image segmentation model to respond to user feedback

This is Part 2 of a series of articles about training image segmentation models so that the models respond to user feedback and adjust their predictions based on the feedback (mouse clicks).

In Part 1 we’ve described the general strategy for training off-the-shelf image segmentation models to respond to user feedback. The problem identified at the end of Part 1 was that manually generating the clicks needed …

image-segmentation machine learning simulated-annealing transformers voronoi-diagrams