UiPath AI Center™

UiPath AI Center™

Image Classification

Out of the Box Packages > UiPath Image Analysis > Image Classification



The Image Classification model is currently in public preview.
UiPath is committed to stability and quality of our products, but preview features are always subject to change based on feedback that we receive from our customers. Using preview features is not recommended for production deployments.

This model runs smoothly on CPU, but you may encounter issues while running on GPU during the preview.

This preview model is a retrainable deep learning model used to classify images. You can train it on your own data and create an ML Skill to perform image classification. This ML Package must be retrained, if deployed without training first, deployment will fail with an error stating that the model is not trained.

Model details

Input type


Input description

Full path of the image file on which you want to classify.
Make sure that the image format is either JPEG or PNG.

Output description

JSON with identified label for the image and confidence score (between 0-1).

  "response": {
    "label": "car",
    "confidence": 0.85657345056533813

Recommend GPU

By default, a GPU is recommended.

Training enabled

By default, training is enabled.


All three types of pipelines (Full Training, Training, and Evaluation) are supported by this package. For most use cases, no parameters need to be specified, the model is using advanced techniques to find a performant model. In subsequent trainings after the first, the model uses incremental learning (that is, the previously trained version will be used, at the end of a Training Run).

Dataset format

For training and evaluation datasets, point to a folder with a subfolder called images and this subfolder can contain input several folders with different classes (for example, a folder called cats with pictures of cats, and another one called dogs with pictures of dogs, and so on).
Example :

-- <Training / Evaluation Directory>
   -- images
      -- Bus
         -- bus001.jpg
         -- bus002.jpg
         -- bus003.jpg
      -- Truck
         -- truck001.jpg
         -- truck012.png
         -- truck0030.jpeg
      -- Car

Environment variables

  • Epochs - default value 20


Classification report

precision    recall  f1-score   support

    Positive       0.75      0.90      0.82        10
    Negative       0.88      0.70      0.78        10

    accuracy                           0.80        20
   macro avg       0.81      0.80      0.80        20
weighted avg       0.81      0.80      0.80        20

Confusion matrix



This is a CSV file with predictions on the test set used for evaluation.

filename      actual        predicted
38  00043.jpg   Positive    Positive
17  00001.jpg   Positive    Positive
59  00014.jpg   Negative    Positive
31  00015.jpg   Positive    Positive
15  00008.jpg   Positive    Positive
69  00025.jpg   Negative    Negative
49  00003.jpg   Positive    Positive
5     00034.jpg Positive    Positive
36  00044.jpg   Positive    Positive
50  00042.jpg   Negative    Positive
96  00011.jpg   Negative    Negative
53  00046.jpg   Negative    Positive
94  00036.jpg   Negative    Negative

Updated 5 months ago

Image Classification

Suggested Edits are limited on API Reference Pages

You can only suggest edits to Markdown body content, but not to the API spec.