View Predictions

After training a model, you can see the model's predictions as overlays on the images in your dataset. The overlay format depends on the project type:

• Object Detection: Predictions are dotted bounding boxes that are the color of the class the model predicted.
  
• Segmentation: Predictions are highlights that are the color of the class the model predicted.
  
• Classification: The predicted class name displays in the bottom left corner of the image. A green check mark means the prediction matches the label. A red "x" means the prediction doesn't match the label.
  

Use Prediction Overlays to Help You Troubleshoot

The predictions give you a visual indication of how your model performed. If the predictions are on top of or close to the labels, then the model performed well. If there's a discrepancy between the label and prediction, find out the root cause; resolving it will likely improve model performance.

For example, in the image below, there's a blue prediction over a yellow label. This could indicate that the object was mislabeled or that the model needs more labeled images of both classes to correctly detect them.

An Image with a Ground Truth and Prediction Mismatch Is a Good Place to Start Troubleshooting

View Predictions for the Dataset

You can see a model's predictions for images in the Build tab. You can also change the applied model and its confidence threshold.

To do this:

1. Open the project to the Build tab.
2. Turn on the Prediction toggle.
3. By default, the most recently used or trained model is applied. If you want to select a different model or adjust the confidence threshold, click the model name.
   Turn on the Prediction Toggle
4. Select the model you want to apply to the dataset from the Model drop-down menu. Models are listed in the chronological order in which they were created.
5. If you want to adjust the confidence threshold, adjust the Confidence Threshold slider. 
   Select the Model and Confidence Threshold You Want to Apply to the Dataset
6. To close the Prediction menu, click anywhere off of the menu.

Confidence Threshold

Images vs. Instances

When the Prediction toggle is on, you can choose to view predictions by image or by instance. These are controlled by the Images and Instances buttons on the Build tab.

Images View

When you select Images, all the images in the dataset display. This is the default view.

Images View

Instances View

When you select Instances, each labeled or predicted area displays as a separate image. You can think of this as LandingLens displaying "one instance" of a label or prediction at a time. You can filter the instances by Ground truth (the label) or Prediction. The images are zoomed in to help you see the object. Instances view is helpful if the objects of interest are too small to see when viewing the dataset in Images view.

Instances View of Ground Truth Labels

Select the Prediction radio button in Instances view to filter by model predictions. The predictions overlay includes the confidence score, which represents how certain (or "confident") the model is that its prediction is accurate. For example, in the screenshot below, the model is 39% confident that the prediction on the left is a Scratch. 

View Each Instance of a Prediction

View Predictions for Each Image

Click an image to open it and see the model's predictions. You can toggle the Labels and Predictions on and off. To learn how to turn predictions into labels, go to Label Assist.

View Predictions and Ground Truth Labels on Each Image

Why Do Some Images Not Have Predictions?

Some images in a dataset may not have predictions. In these situations, it's because the model did not have an opportunity to make predictions on those images. For example, if Predict at Upload is disabled and you upload an image after training a model, the model doesn't have an opportunity to make a prediction on that image.

The Image on the Left Doesn't Have a Prediction Because it Was Uploaded After Model Training & "Predict at Upload" Was Off

When Does a Model Make Predictions?

When you train a model, the model makes predictions for the images currently in the dataset. But there are other situations that also trigger a model to make predictions on an image. Here is a list of all situations that cause a model to make predictions on an image:

• When you train a model, the model makes predictions on all images currently in the dataset.
• If Predict at Upload is enabled and you upload an image after training a model, the model makes a prediction on the image.
  
  Predict at Upload
• If you go to the Models page and run a model on an evaluation set, the model checks what images it hasn't "seen" before. If the model hasn't "seen" the image before, it makes a prediction on it. You can then view those predictions when you select that model on the Build tab.
  
  When the Model Was Evaluated on "Snapshot-06-27-2024 10:31", It Made Predictions on Any Images It Hadn't "Seen" Before
• If you open an image and click Get Prediction, the selected model makes a prediction on the image.
  
  When You Click "Get Prediction", the Model Makes a Prediction on the Image