The ClassificationAgent
performs image classification.
ClassificationAgent
is initialized with two arguments:
classes
list.ClassificationModel
. If left unspecified, this parameter defaults to the CLIP()
Model.The supported ClassificationModel
models can be found below:Note: Since a text embedding of each class will be used to when classifying the image, a more specific class name will yield a more specific detection. You should scope your class names accordingly. For instance, a"car"
class will result in more general detections than a"blue Honda Accord"
class.