모델 가중치 다운로드

자체 하드웨어에서 Roboflow 모델을 실행하려면 Roboflow Inference(권장되는 자동 방법)를 사용하거나 특정 엣지 케이스를 위해 모델 가중치를 수동으로 다운로드할 수 있습니다.

Roboflow Inference를 통한 자동 다운로드(권장)

Roboflow Inference 로컬의 CPU 및 GPU 장치에서 모델을 실행하기 위한 당사의 오픈 소스 확장 가능 시스템입니다.

이것은 시작하는 가장 빠르고 신뢰할 수 있는 방법입니다 Roboflow Inference를 사용하면 파일이나 버전 관리를 직접 할 필요가 없습니다; Roboflow Inference는 코드를 처음 실행할 때 모델 가중치를 자동으로 가져와 캐시합니다.

• 작동 방식: 첫 번째 추론 요청 시 가중치는 Roboflow의 서버에서 다운로드되어 로컬에 저장됩니다. 이후 모든 예측은 이 로컬 캐시를 사용합니다 — 이미지가 클라우드로 전송되지는 않습니다.

• 배포 옵션:

  • Workflows: 프로덕션 준비가 된 다단계 컴퓨터 비전 워크플로우용

  • Python inference SDK: Python 통합용

수동 모델 가중치 다운로드

때때로 원시 가중치 파일(예: PyTorch .pt 파일)이 필요할 수 있으며, 예를 들어 맞춤형 Android 구현과 같이 Roboflow가 아직 네이티브로 지원하지 않는 장치에서 실행하려는 경우에 해당합니다.

다음을 참조하세요 지원 모델 표 에서 가중치 다운로드 호환성을 확인하세요.

프리미엄 기능: 수동 가중치 다운로드는 Core 요금제의 유료 사용자와 특정 Enterprise 고객에게만 제공됩니다. 자세한 내용은 당사의 요금 페이지.

방법 A: Roboflow 플랫폼

프로젝트 내의 모델 버전으로 이동하세요. 요금제가 허용하는 경우 "Download Weights" 버튼을 클릭하면 가중치를 다운로드할 수 있습니다. 이렇게 하면 임베디드 장치에서 사용하기 위해 변환할 수 있는 파일이 제공됩니다.

방법 B: Python SDK

또한 Roboflow Python 패키지를 사용하여 가중치를 직접 디렉토리로 다운로드할 수 있습니다:

import roboflow

rf = roboflow.Roboflow(api_key="YOUR_API_KEY")
model = rf.workspace().project("PROJECT_ID").version("1").model
model.download() # 'weights.pt'를 로컬 폴더로 다운로드합니다

참고: Roboflow는 Roboflow Inference 생태계 외부에서 사용되는 모델 가중치에 대한 기술 지원을 제공하지 않습니다. 최상의 경험을 위해 섹션 1에 설명된 Inference 경로를 사용하는 것을 권장합니다.