Classification
Adding classification datasets.

Uploading Classification Datasets

If you are looking to label entire images as belonging to a class, then you will need to upload a classification dataset. Classification datasets require images/videos and in distinct folders. Class names are derived based on the folder names.
  • For example, if you are uploading images of dogs, cats, and raccoons, you should have three folders.
    • One folder should be called "dogs" and contain all dog images.
    • Another folder should be called "cats" and contain all cat images.
    • The third folder should be called "raccoons" and contain all raccoon images.
  • For detection of a single item in a Classification dataset, select Single-Label Classification
  • For detection of multiple items in a Classification dataset, select Multi-Label Classification
Note that if the goal is to detect movement or position, it is better to create an object detection project with the desired data. A Classification project merely answers the question "Is this label or annotation present in this image?"

Models

You can use these tutorials to fit classification models to your own datasets. If you want to explore one-click AutoML for classification, reach out to us.

Converting Object Detection Data Into Classification Data

Have some data in a Roboflow object detection project that you want to use in a classification project? Here's how:
  • Generate a dataset in the source project with the isolate preprocessing step, but no augmentation options, selected.
    • This will create crops of all the bounding boxes from the projects dataset for a classifier to be trained on
  • Export the data from that version using the OpenAI Clip Classification method (see screenshot) and download to your local machine.
  • Unzip the downloaded data, create a Roboflow classification project, and upload the unzipped folder of images to it
  • Review the data, generate a new dataset, and train!

Images

We have made available a few image classification datasets.

Blog Posts

We regularly write and share blog posts, including tutorials, recommendations for preprocessing and augmenting data, and techniques for improving model performance. Check out our computer vision modeling-related blog posts here and all blog posts here.