from ultralytics import YOLO

# 사전학습 모델 로드
model = YOLO('yolo11n.pt')  # nano (가장 가벼운 모델)

# 모델 정보 확인
model.info()

# 이미지 추론
results = model('image.jpg')

# 결과 확인
for result in results:
    boxes = result.boxes
    for box in boxes:
        cls = int(box.cls)
        conf = float(box.conf)
        xyxy = box.xyxy[0].tolist()
        print(f"클래스: {cls}, 신뢰도: {conf:.2f}, 좌표: {xyxy}")

# 모델 크기 옵션 (n < s < m < l < x)
models = ['yolo11n.pt', 'yolo11s.pt', 'yolo11m.pt', 'yolo11l.pt', 'yolo11x.pt']