Image Classification
A guide to AutoML image classification models
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 | Latency | Contains Hive Classes | Use case | Dataset Validation |
|---|---|---|---|---|
| Hive Image Classification | Low | No | If you're looking to build a custom image classification model from scratch, choose this base model. | 1. Each class requires a minimum of 10 examples. 2. Each image file shouldn't be larger than 50MB 3. Each row of image url character length shouldn't be longer than 2048 characters. |
| Hive Vision Moderation | Low | Yes | If you're interested in moderation use cases, you should select this base model. This option will fine-tune our existing Hive Visual Moderation model with the additional data you provide, allowing you to customize our pre-made heads or add in your own new ones. The resulting model will include all Hive Visual Moderation heads as well as any you choose to create. | 1. Each class requires a minimum of 50 examples. 2. Each image file shouldn't be larger than 50MB 3. Each row of image url character length shouldn't be longer than 2048 characters. |
Labels and Classes for Hive Vision 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 these pre-made labels is shown below:
| Labels | Classes |
|---|---|
| nsfw | general_not_nsfw_not_suggestive general_nsfw general_suggestive |
| female_underwear | no_female_underwear yes_female_underwear |
| male_underwear | no_male_underwear yes_male_underwear |
| sex_toy | no_sex_toy yes_sex_toy |
| female_nudity | no_female_nudity yes_female_nudity |
| male_nudity | no_male_nudity yes_male_nudity |
| female_swimwear | no_female_swimwear yes_female_swimwear |
| male_shirtless | no_male_shirtless yes_male_shirtless |
| text | no_text text |
| animated | animated hybrid natural |
| gun | animated_gun gun_in_hand gun_not_in_hand no_gun |
| knife | culinary_knife_in_hand culinary_knife_not_in_hand knife_in_hand knife_not_in_hand no_knife |
| blood | a_little_bloody no_blood other_blood very_bloody |
| fight | no_fight yes_fight |
| pills | no_pills yes_pills |
| smoking | no_smoking yes_smoking |
| injectables | illicit_injectables medical_injectables no_injectables |
| nazi | no_nazi yes_nazi |
| kkk | no_kkk yes_kkk |
| middle_finger | no_middle_finger yes_middle_finger |
| terrorist | no_terrorist yes_terrorist |
| texture | no_texture yes_texture |
| snapcode | no_snapcode yes_snapcode |
| booty | booty_exposed_butt booty_exposed_butt_and_head booty_none |
| scribble | no_scribble scribble |
| overlay_text | no_overlay_text yes_overlay_text |
| sexual_activity | no_sexual_activity yes_sexual_activity |
| marijuana | no_marijuana yes_marijuana |
| vaping | no_vaping yes_vaping |
| tobacco | no_tobacco yes_tobacco |
| military_conflict | no_military_conflict yes_military_conflict |
| hanging | hanging no_hanging_no_noose noose |
| realistic_nsfw | no_realistic_nsfw yes_realistic_nsfw |
| corpse | animated_corpse human_corpse no_corpse |
| self_harm | no_self_harm yes_self_harm |
| drawing | no_drawing yes_drawing |
| emaciated_body | no_emaciated_body yes_emaciated_body |
| child_present | no_child_present yes_child_present |
| sexual_intent | no_sexual_intent yes_sexual_intent |
| animal_genitalia | animal_genitalia_and_human animal_genitalia_only animated_animal_genitalia no_animal_genitalia |
| blurry | a_little_blurry blurry no_blur very_blurry |
| pre_teen | no_pre_teen yes_pre_teen |
| gambling | no_gambling yes_gambling |
| undressed | no_undressed yes_undressed |
| confederate | no_confederate yes_confederate |
| alcohol | animated_alcohol no_alcohol yes_alcohol yes_drinking_alcohol |
| bra | no_bra yes_bra |
| sports_bra | no_sports_bra yes_sports_bra |
| panties | no_panties yes_panties |
| cleavage | no_cleavage yes_cleavage |
| sportswear_bottoms | no_sportswear_bottoms yes_sportswear_bottoms |
| negligee | no_negligee yes_negligee |
| miniskirt | no_miniskirt yes_miniskirt |
| bodysuit | no_bodysuit yes_bodysuit |
| religious_icon | no_religious_icon yes_religious_icon |
| watermark | no_watermark yes_watermark |
| qr_code | no_qr_code yes_qr_code |
| child_abuse | no_child_abuse yes_child_abuse |
| animal_abuse | no_animal_abuse yes_animal_abuse |
| bulge | no_bulge yes_bulge |
You can find a more detailed list of these classes and their definitions here.
Updated about 1 month ago