YOLOv9

Serverless Hosted API를 통해 YOLOv9 객체 탐지를 사용합니다.

당사는 YOLOv9 객체 감지 추론을 지원합니다 Serverless Hosted API. YOLOv9 학습은 Roboflow에서 지원되지 않지만, 다음을 할 수 있습니다 사전 학습된 가중치를 업로드 기존 Project에 대해 업로드하고 Serverless Hosted API를 통해 제공할 수 있습니다.

자체 호스팅 배포는 다음을 참조하세요 Roboflow Inference.

YOLOv9 입력 크기는 Roboflow 외부에서 모델을 학습할 때 설정됩니다(일반적인 값: 640x640 또는 1280x1280).

코드 샘플

1

API Key를 받으세요

Roboflow 계정을 만들고, 다음에서 키를 찾으세요: Roboflow API 설정 페이지 그리고 이를 셸에서 사용할 수 있도록 설정하세요:

export ROBOFLOW_API_KEY="your-key-here"

2

종속성을 설치하세요

SDK를 설치하고 supervision 디코딩 및 주석 달기를 위해:

pip install inference-sdk supervision

3

모델을 실행하세요

이 예제는 다음에서 학습된 Roboflow의 공개 YOLOv9 모델을 실행합니다 COCO (coco/18), 따라서 그대로 작동합니다. 자신의 가중치를 제공하려면, 다음으로 바꾸세요 {workspace}/{model-slug} ID로 호출할 수 있습니다(참조 Versions, Trainings, and Models).

import os
import cv2
import numpy as np
import requests
import supervision as sv
from inference_sdk import InferenceHTTPClient

content = requests.get("https://media.roboflow.com/quickstart/traffic.jpg").content
image = cv2.imdecode(np.frombuffer(content, np.uint8), cv2.IMREAD_COLOR)

client = InferenceHTTPClient(
    api_url="https://serverless.roboflow.com",
    api_key=os.environ["ROBOFLOW_API_KEY"],
)
result = client.infer(image, model_id="coco/18")

detections = sv.Detections.from_inference(result)

annotated = sv.BoxAnnotator().annotate(image.copy(), detections)
annotated = sv.LabelAnnotator().annotate(annotated, detections)

cv2.imwrite("annotated.png", annotated)

설정 api_url 을 배포 대상에 맞게 설정하세요:

• https://serverless.roboflow.com Serverless Hosted API용입니다.

• http://localhost:9001 로컬 Inference 서버용입니다.

• 귀하의 Dedicated Deployment 비공개 엔드포인트용 URL입니다.