1. A device comprising: a processor; and a computer-readable medium storing instructions that, when executed by the processor, cause the processor to perform acts comprising: receiving a training image; generating candidate segment masks based at least in part on the training image; ranking the candidate segment masks to generate ranked candidate segment masks; selecting a set of the ranked candidate segment masks to generate a set of ranked candidate segment masks; selecting one of the set of ranked candidate segment masks to train the neural network; and training the neural network.