jelly_image.py

import os
from ultralytics import YOLO

def predict_jelly_image(image_path):
    model_path = os.path.join(os.path.dirname(__file__), "models", "image_best.pt")
    
    model = YOLO(model_path)
    
    results = model.predict(source=image_path, save=False)
    
    predictions = []
    for result in results:
        for conf, cls in zip(result.boxes.conf, result.boxes.cls):
            class_name = result.names[int(cls)]
            confidence_percent = conf.item() * 100
            predictions.append({
                'class': class_name,
                'confidence': confidence_percent
            })
    return predictions

if __name__ == '__main__':
    print(predict_jelly_image('노무라입깃.jpg'))