분류

Roboflow에 호스팅된 분류 모델에서 추론을 실행하세요.

로컬 및 호스팅된 이미지에서 추론

종속 항목을 설치하려면, pip install inference-sdk.

from inference_sdk import InferenceHTTPClient

CLIENT = InferenceHTTPClient(
    api_url="https://classify.roboflow.com",
    api_key="API_KEY"
)

result = CLIENT.infer(your_image.jpg, model_id="vehicle-classification-eapcd/2")

응답 객체 형식

Single-Label Classification

호스팅된 API 추론 경로는 다음을 반환합니다 JSON 예측 배열을 포함하는 객체를 반환합니다. 각 예측은 다음 속성을 가집니다:

  • time = 이미지를 처리하고 예측을 반환하는 데 걸린 총 시간(초)

  • 이미지 = 이미지에 대한 정보를 담고 있는 객체 width 이제 GPU TRT 컨테이너가 Docker에서 실행되고 있습니다. 다른 Ubuntu 터미널을 열어 Docker 컨테이너로 추론 데이터를 보낼 수 있습니다. 다음 명령을 사용하세요: height

    • width 예측된 이미지의 높이

    • height = 예측된 이미지의 높이

  • predictions = 예측에 대한 모든 예측 클래스와 해당 신뢰도 값을 모아 놓은 컬렉션

    • class = 분류의 레이블

    • confidence = 이미지에 감지된 분류의 객체가 포함되어 있을 확률에 대한 모델의 신뢰도

  • top = 가장 높은 신뢰도의 예측 클래스

  • confidence = 가장 높은 예측 신뢰도 점수

  • image_path = 예측된 이미지의 경로

  • prediction_type = 추론을 수행하는 데 사용된 모델 유형, ClassificationModel 이 경우

// 예시 JSON 객체
{
  "time": 0.19064618100037478,
  "image": {
    "width": 210,
    "height": 113
  },
  "predictions": [
    {
      "class": "real-image",
      "confidence": 0.7149
    },
    {
      "class": "illustration",
      "confidence": 0.2851
    }
  ],
  "top": "real-image",
  "confidence": 0.7149,
  "image_path": "/cropped-images-1.jpg",
  "prediction_type": "ClassificationModel"
}

API 참조

Inference API 사용하기

POST https://classify.roboflow.com/:datasetSlug/:versionNumber

base64로 인코딩된 이미지를 모델 엔드포인트로 직접 POST할 수 있습니다. 또는 이미지가 이미 다른 곳에 호스팅되어 있다면 쿼리 문자열의 이미지 매개변수로 URL을 전달할 수 있습니다.

경로 매개변수

이름
유형
설명

datasetSlug

string

데이터셋 이름의 URL 안전 버전입니다. 웹 UI에서 메인 프로젝트 뷰의 URL을 확인하면 찾을 수 있습니다.

string

데이터셋 버전을 식별하는 버전 번호입니다.

쿼리 매개변수

이름
유형
설명

api_key

string

작업공간 API 설정 페이지에서 얻은 귀하의 API 키

{
   "predictions":{
      "bird":{
         "confidence":0.5282308459281921
      },
      "cat":{
         "confidence":0.5069406032562256
      },
      "dog":{
         "confidence":0.49514248967170715
      }
   },
   "predicted_classes":[
      "bird",
      "cat"
   ]
}

Last updated

Was this helpful?