image-classification

AI Image Classification model for sorting my photos

Source: https://huggingface.co/ultralyticsplus/yolov8s

Uses the Yolo8 model for classification.

pre-requisites:

• install ultralyticsplus: pip install -U ultralyticsplus==0.0.14

load model and preform prediction: from ultralyticsplus import YOLO, render_result

load model

model = YOLO('ultralyticsplus/yolov8s')

set model parameters

model.overrides['conf'] = 0.25 # NMS confidence threshold model.overrides['iou'] = 0.45 # NMS IoU threshold model.overrides['agnostic_nms'] = False # NMS class-agnostic model.overrides['max_det'] = 1000 # maximum number of detections per image

set image

image = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'

perform inference

results = model.predict(image)

observe results

print(results[0].boxes) render = render_result(model=model, image=image, result=results[0]) render.show()