Image Classification

Image classification is the task of automatically categorizing images into predefined classes or categories based on their visual content.

input

IMAGE

LABEL

output

LABEL

area

Code

from grandmaster import predict

predict(
    task_name="image-classification",
    image="https://grandmaster.ai/ex/puppy.jpeg",
    candidate_labels=["god", "cat"],
)

Input schema

image : valid URL containing an image

candidate_labels : list of string to classify

Output schema

list of dictionaries of type:

label : label name

score : probability of the label