Image Classification - AutoML Training
A guide on training your custom image classification models using Hive AutoML
Overview
Image classification models can be used to categorize images and videos. Some popular use cases for this model type are object classification, brand identification, and vision moderation.
Which Base Model Should I Use?
| Base Model | Use Case | Hive Classes | Dataset Requirements |
|---|---|---|---|
| Image Classification v2 | Well-suited for most image classification tasks | -- | Minimum 10 examples per class |
| Visual Moderation v2 | Best for moderation tasks—leverages pre-trained Hive moderation labels | Included | Minimum 50 examples per class |
Labels and Classes for Hive Visual Moderation
If you select the Hive Visual Moderation base model, the model output will always contain all Hive Visual Moderation categories. A full list of moderation labels can be found in the Visual Moderation documentation.
Updated 9 days ago