AiVibe

AI & Machine Learning

Image Classification

Image classification is the computer vision task of assigning one or more labels to an entire image, such as good or defective, or the type of component shown, without locating where in the image the object or defect is.

A classifier, typically a convolutional neural network or vision transformer, outputs a probability for each class. Single-label classification chooses one class, while multi-label classification can assign several. Models are usually created by fine-tuning a network pre-trained on a large dataset such as ImageNet, whose annual recognition challenge drove much of the progress in deep learning for vision.

In manufacturing, classification sorts products, verifies part types, grades surface quality and makes pass or fail decisions when the image is tightly framed around one item. It is also used for recognising document types and sorting waste or recyclables.

Because it gives no location, classification is unsuitable when several objects appear or when the position or size of a defect matters; detection or segmentation is then needed. Class imbalance is common in inspection, where defects are rare, so per-class precision and recall matter more than overall accuracy. Confidence thresholds should be set using validation data.

Key points

Where AiVibe comes in

AiVibe delivers AI and machine learning services, chatbots and virtual assistants with RAG, MCP tools and voice, AI quality management including bias detection and model validation, and the AIMURUGA AI agent, and builds Intel-based edge AI devices using the Intel Distribution of OpenVINO toolkit.

Explore AiVibe’s work in AI & Machine Learning →

Related terms

Ask AiMuruga can explain Image Classification for your plant, product or security programme, and draw how it fits.