Subject matter regards improving image segmentation or image annotation. A method can include receiving, through a user interface (UI), for each class label of class labels to be identified by the ML model and for a proper subset of pixels of the image data, data indicating respective pixels associated with the class label, partially training the ML model based on the received data, generating, using the partially trained ML model, pseudo-labels for each pixel of the image data for which a class label has not been received, and receiving, through the UT, a further class label that corrects a pseudo-label of the generated pseudo-labels.