Model-Assisted Labeling

Reduce annotation time and get a model into production faster by using model weights from a trained model to label your data

Model-assisted labeling allows you to use a previous version of a trained model or a public model on Roboflow Universe as an annotation assistant. Model-assisted labeling is ideal for identifying specific classes in an image automatically.

When model-assisted labeling is enabled, the selected model will run when you open an image in the annotation tool. The model will predict objects in the image. Annotations are added for each object the model finds.

How to Use Model-Assisted Labeling

Open an image in Roboflow Annotate then click the magic wand icon in the command tray.

A pop-up will appear asking what model you want to use to help you label images:

Select a model from the pop up. You can select a model to which you have access through the Your Models tab. You can select any model from Universe through the Public Models tab. You need to star a model on Universe before it will show up in the Public Models tab.

Click "Continue" once you have selected a model.

Next, you will be asked what classes you want to find. By default, the selected model will return predictions for all classes. You can uncheck classes the model should return from the pop up.

If you want to remap any class names, you can do so. This is useful if the model you are using has different class names from the classes you want to add to images in your dataset.

Once you have configured the classes for use with the selected model, click the "Select Classes" button. You will then be able to start Label Assist.

Label Assist will start running on the current image you are annotating. Label Assist will persist as you navigate through images in your dataset using the forward and back arrows in the annotation interface.

Here is an example of Label Assist recommending an annotation: