모델 학습

Roboflow 대시보드에서 최신 기술을 사용해 모델을 학습하세요.

소개

Roboflow 인터페이스에서 컴퓨터 비전 모델을 학습할 수 있습니다.

Roboflow는 두 가지 학습 옵션을 제공합니다:

• Roboflow Custom Train: 프로덕션에 바로 사용할 수 있는 모델을 만드는 데 이상적인 우리의 대표 학습 서비스입니다.

• 신경망 아키텍처 탐색: 새로운 모델 아키텍처를 발견하고 동시에 미세 조정하세요.

또한, 첫 번째 이미지 주석 배치를 승인하면, 하나의 Roboflow Instant Model 은 자동으로 학습됩니다. 이러한 모델은 자동 레이블링이나 배포에 즉시 사용할 수 있습니다.

Roboflow에서 학습된 모델은 온디바이스 추론 서버인 Inference를 사용하거나, Serverless Cloud API와 Workflows, Workflows를 이용한 Batch Processing, 또는 모델 API 엔드포인트를 통해 클라우드에 배포할 수 있습니다.

자세한 내용은 라이선스 안내 를 읽고 Roboflow에서 학습된 모델의 라이선스에 대해 자세히 알아보세요.

웹 앱

모델 학습

컴퓨터 비전 모델을 학습하려면 프로젝트 탐색에서 "Train"을 방문하세요.

"Custom Training" 버튼을 클릭하여 학습 작업 구성을 시작하세요:

신경망 아키텍처 탐색(NAS)

객체 탐지 및 인스턴스 분할 프로젝트에서는 NAS를 신경망 아키텍처 탐색(NAS) 단일 모델 아키텍처를 선택하는 대신 사용할 수 있습니다. NAS는 자동으로 여러 모델 구성을 학습하고 평가한 다음, 정확도와 지연 시간 요구 사항을 기반으로 최적의 모델을 추천합니다.

NAS를 사용하려면 학습 엔진을 선택할 때 NAS 옵션을 선택하세요. NAS에는 최소 15개의 검증 이미지가 필요합니다.

NAS 학습 차트에 대한 자세한 내용은 학습 결과 보기.

모델 아키텍처 선택

다음으로, 모델 아키텍처와 크기를 선택해야 합니다. 이는 모델을 학습하는 데 사용되는 머신 러닝 기술입니다.

학습할 수 있는 모델 아키텍처는 설정한 프로젝트 유형에 따라 다릅니다. 자세한 내용은 지원되는 모델 표 를 참조하세요.

객체 탐지의 경우 RF-DETR이 최고의 정확도를 제공합니다. 인스턴스 분할의 경우 RF-DETR Seg(미리 보기)가 최고의 정확도를 제공합니다. 키포인트 탐지의 경우 RF-DETR Keypoint(미리 보기)도 사용할 수 있습니다.

모델 크기는 선택한 모델의 아키텍처에 따라 달라집니다. 예를 들어 최첨단 객체 탐지 모델인 RF-DETR은 Nano, Small, Medium, Base를 제공합니다. Medium, Large, Extra Large는 유료 사용자에게만 제공됩니다. 다음을 학습하는 것은 SAM3 모델은 다음이 포함된 유료 요금제에서 사용할 수 있습니다 사용량 기반 청구. 그러면 SAM3가 아키텍처 목록에 "Request Feature" 버튼과 함께 표시되며, 이를 사용해 접근 권한을 얻을 수 있습니다.

프로젝트 유형에서 사용할 수 있는 아키텍처를 선택한 다음 "Continue"를 클릭하세요:

체크포인트 선택

학습 옵션을 선택한 후 체크포인트에서 학습할지 여부를 묻게 됩니다. 아래 탭에는 각 모델 유형별 구성 옵션이 표시됩니다.

객체 탐지

세 가지 옵션이 있습니다:

• 이전 체크포인트에서 학습: 이미 작동하는 모델이 있고 이를 개선하고 싶을 때 이상적입니다.

• 공개 체크포인트에서 학습: 첫 번째 모델이거나 이전 학습 실행이 예상한 결과를 얻지 못했을 때 이상적입니다.

• 무작위 초기화에서 학습: 고급 사용자만 사용 가능, 이 옵션은 학습을 시작할 빈 상태를 제공합니다. 대부분의 사용자는 이 옵션을 사용할 때 더 나쁜 결과를 봅니다.

분류/시맨틱 분할

분류 및 시맨틱 분할 모델의 경우 사용할 수 있는 체크포인트는 하나뿐입니다.

학습 옵션은 어떻게 선택하나요?

새로운 객체 탐지 프로젝트에는 공개 체크포인트에서 학습하는 것을 권장합니다. 기본적으로 Microsoft COCO 데이터셋으로 학습된 모델에서 학습할 수 있습니다. 분류 및 시맨틱 분할의 경우 ImageNet에서만 학습하는 것을 지원합니다.

Universe에 호스팅된 프로젝트를 기반으로 한 체크포인트에서 학습할 수 있습니다(객체 탐지만 해당). 이렇게 하려면 먼저 Universe에서 프로젝트에 별표 표시를 하고. 그러면 해당 프로젝트는 Roboflow 웹 애플리케이션에서 학습 체크포인트로 사용할 수 있습니다.

또한 이전 버전의 모델을 기반으로 한 체크포인트에서 학습할 수 있습니다(객체 탐지, 인스턴스 분할, 키포인트 탐지만 해당). 이 방법은 더 빠른 학습 프로세스를 가능하게 합니다. 모델이 강력한 성능을 보일 경우에만 모델에 대해 이전 체크포인트에서 학습하는 것을 권장합니다).

체크포인트에서 학습한다는 것은 전이 학습을 사용한다는 의미입니다. 전이 학습은 선택한 모델에서 학습을 초기화합니다. 이는 학습 시간을 줄이고 학습 점수를 개선하는 데 도움이 될 수 있습니다.

처음부터 학습한다는 것은 당신이 전이 학습을 사용하지 않는다는 뜻입니다. 이렇게 하면 모델 가중치의 초기값이 무작위로 설정되어 모델 학습이 시작됩니다.

크레딧 상한 설정

한 번의 학습 실행이 사용할 크레딧 수를 제한할 수 있습니다. 학습 옵션의 "Credit Cap"에 숫자를 입력하거나, 상한 없이 실행하려면 비워 두세요. 상한을 사용하려면 다음이 포함된 요금제가 필요합니다 사용량 기반 청구.

실행이 상한에 도달하면 Roboflow는 이를 중지하고 모델 가중치를 유지합니다. 이는 학습 조기 중지와 동일합니다. 작업이 실행되는 동안, 학습은 사용한 상한 크레딧 수를 표시합니다.

각 모델에는 입력 아래에 표시된 최소 상한이 있습니다. 최소값은 모델을 생성할 수 있는 가장 짧은 실행의 비용이므로 더 작은 상한은 거부됩니다. 비용 추정치보다 낮은 상한은 허용되지만, 실행은 완료되기 전에 중지될 가능성이 큽니다.

신경망 아키텍처 탐색 NAS는 학습과 아키텍처 탐색을 별도로 청구하므로 "Training Credit Cap"과 "Search Credit Cap" 두 개의 상한이 있습니다. 하나를 설정하면 다른 단계에는 상한이 적용되지 않습니다.

상한이 있는 실행이 크레딧을 남기고 중지되면 계속 진행할 수 있습니다. 새 실행은 상한 아래에 남은 크레딧을 사용할 수 있습니다. 남은 크레딧이 모델의 최소값보다 적으면 대신 새 학습을 시작하세요.

학습 작업 시작

학습할 체크포인트를 선택했으면 "Start Training"을 클릭하세요.

그러면 데이터셋이 압축되어 Roboflow 클라우드에서 학습할 수 있도록 준비됩니다.

학습이 시작되기 전에 학습 요약에 예상 기간과 크레딧 비용이 표시됩니다:

데이터셋이 클수록, 그리고 데이터셋의 이미지가 클수록 모델 학습에 더 오래 걸립니다.

학습 과정이 끝나면 이메일을 받게 됩니다. 대부분의 경우 24시간 이내여야 합니다.

가격

Roboflow에서의 학습 가격은 학습 작업의 길이에 따라 책정됩니다. 자세한 내용은 크레딧 페이지.

를 참고하세요. 학생이나 연구자이고 현재 진행 중인 프로젝트에 필요한 크레딧이 있다면, 추가 크레딧을 신청할 수 있습니다.

Agent에서 학습하기

다음에서도 동일한 학습 실행을 시작할 수 있습니다 Roboflow Agent 채팅을 떠나지 않고도 가능합니다. Agent에서 프로젝트 탭을 열고 "Train"을 클릭하세요. 학습 단계는 탭 안에서 열립니다: 학습 엔진을 선택한 다음 아키텍처와 크기를 선택하고, 그다음 데이터셋 버전을 선택하세요.

선택한 버전이 아직 내보내지지 않았다면 Agent가 먼저 내보내기를 준비하고 진행 상황을 표시합니다. 내보내기가 준비되면 학습이 시작됩니다. 프로젝트에 버전이 없다면 같은 화면에서 생성할 수 있습니다.

이 방식으로 학습하려면 Train Model 및 List Versions 권한이 필요합니다.

한 버전에서 여러 모델 학습

동일한 데이터셋 버전에서 하나 이상의 모델을 학습시켜 동일한 데이터에 대해 아키텍처나 크기를 비교할 수 있습니다. 각 학습은 독립적으로 실행되고 평가되며, 버전 페이지에는 모든 학습이 나열되어 전환할 수 있습니다.

HTTP API

프로젝트에 생성된 데이터셋 버전이 있으면, 호스팅된 모델을 생성하는 학습 작업을 예약할 수 있습니다.

학습 작업 시작

데이터셋 버전에서 학습 실행을 시작합니다. 데이터셋 버전은 URL 경로의 일부입니다.

POST https://api.roboflow.com/:workspace/:project/:version/train

본문

이름	유형	설명	필수
model_type	문자열	아키텍처 식별자(예: rfdetr-nano, yolov8, yolov11). 잘못된 값을 전달하면 전체 목록이 오류로 반환됩니다.
checkpoint	문자열	학습을 이어갈 선택적 체크포인트입니다.
epochs	정수	에포크 수.
speed	문자열	빠름 (기본값) 또는 정확함. 정확함 은 유료 기능입니다.
trainingCreditCap	숫자	학습이 중지되기 전에 사용할 수 있는 최대 크레딧 수입니다. 최소값 이상이어야 합니다.
nasMiningCreditCap	숫자	NAS 아키텍처 탐색이 사용할 수 있는 최대 크레딧 수입니다. NAS 모델 유형에서만 유효합니다.

curl "https://api.roboflow.com/my-workspace/my-detector/3/train?api_key=$ROBOFLOW_API_KEY" \\
  -X POST \\
  -H "Content-Type: application/json" \\
  -d '{"model_type": "rfdetr-base"}'

학습 실행은 비동기적으로 진행됩니다. 응답은 작업이 큐에 추가되었음을 확인할 뿐, 완료되었음을 의미하지는 않습니다. 진행 상황을 확인하려면 학습 결과 를 폴링하거나 버전의 모델 필드에서 학습된 모델이 나타나는지 확인하세요.

버전이 아직 학습할 준비가 되지 않았다면, 요청은 400 를 코드 와 해결 방법 필드를 포함해 반환합니다. 해결 방법은 수정 사항의 이름을 알려주므로 스크립트나 에이전트가 메시지를 읽을 필요가 없습니다: 재균형 (이미지는 충분하지만 잘못된 분할에 있음), 업로드 (데이터셋에 더 많은 이미지가 필요함), 주석 추가 (이미지에 레이블이 필요함), 재생성 (버전 생성이 실패했으므로 새 버전을 생성하세요), 또는 대기 (재균형 또는 생성이 아직 실행 중임). 데이터와 관련이 없는 오류, 예를 들어 플랜 제한과 같은 경우에는 해결 방법이 없습니다. 해결 방법.

플랜 제한은 웹 앱과 동일한 방식으로 여기에 적용됩니다. 요금제에 요청한 아키텍처가 포함되어 있지 않으면, 요청은 400 를 반환하고 작업은 시작되지 않습니다: nas_not_available_for_plan 다음 NAS 모델 유형에는, sam3_not_available_for_plan 다음 SAM3 모델 유형이 해당됩니다.

모델의 최소값보다 낮은 크레딧 상한, 또는 크레딧으로 청구하지 않는 작업 공간의 모든 상한은 400 코드와 함께 invalid_credit_cap를 반환합니다. 오류 메시지에 최소값이 표시됩니다.

호출하는 API 키에는 trainingJob:create 스코프가 있어야 합니다.

Python SDK

Version.train() 은 Roboflow 플랫폼에서 학습을 예약합니다. 이 호출은 작업이 큐에 추가되면 반환되며, 학습 자체는 비동기적으로 실행됩니다.

import roboflow

rf = roboflow.Roboflow(api_key="YOUR_API_KEY")
project = rf.workspace().project("my-detector")

# 아직 없다면 전처리 및 증강이 포함된 버전을 만듭니다.
new_version = project.generate_version({
    "preprocessing": {
        "auto-orient": True,
        "resize": {"width": 640, "height": 640, "format": "Stretch to"},
    },
    "augmentation": {},
})
version = project.version(new_version)

# 학습을 예약합니다.
model = version.train(
    model_type="rfdetr-nano",   # 잘못된 값을 전달하면 전체 목록이 오류로 반환됨
    checkpoint=None,             # 선택 사항: 이전 체크포인트에서 재개
    epochs=100,                  # 선택 사항: 기본값은 모델 유형에 따라 다름
    plot_in_notebook=False,      # 학습 진행 그래프 표시(노트북 전용)
)

매개변수

• model_type (str) - 아키텍처 식별자. 일반적인 값: rfdetr-nano, rfdetr-base, yolov8, yolov11. 프로젝트 유형에 따라 다릅니다. 잘못된 값을 전달하면 전체 목록이 오류로 반환됩니다.

• speed (str, 선택 사항) - "fast" (기본값) 또는 "accurate". 정확한 학습은 유료 기능입니다.

• checkpoint (str, 선택 사항) - 재개할 체크포인트의 ID.

• epochs (int, 선택 사항) - 에포크 수. 기본값은 모델 유형에 따라 다릅니다.

• plot_in_notebook (bool, 기본값 False) - 학습 진행 상황을 인라인으로 표시(노트북 전용).

학습 후

학습이 완료되면, Version의 .model 속성은 호스팅된 모델을 반환합니다:

predictions = version.model.predict("photo.jpg", confidence=40, overlap=30).json()

참고 이미지에서 모델 실행 에서 전체 추론 참조를 확인하세요.

CLI

명령줄에서 모델 학습을 시작할 수 있습니다. 학습은 Roboflow의 클라우드 인프라에서 실행됩니다.

명령

roboflow train start -p <project> -v <version> -t <model-type>

또는 축약형을 사용하세요( start):

roboflow train -p <project> -v <version> -t <model-type>

옵션

플래그	설명
-p, --project	프로젝트 ID(필수)
-v, --version	버전 번호(필수)
-t, --type	모델 유형, 예: yolov8n, rfdetr-nano
--checkpoint	학습을 재개할 체크포인트
--speed	학습 속도 프리셋
--epochs	학습 에포크 수

예시

특정 모델 유형으로 학습을 시작:

roboflow train start -p my-detector -v 3 -t yolov8n

체크포인트에서 재개:

roboflow train start -p my-detector -v 3 -t yolov8n --checkpoint my-checkpoint

JSON 출력

roboflow train start -p my-detector -v 3 -t yolov8n --json

{
  "status": "training_started",
  "project": "my-detector",
  "version": 3
}

다음 단계

• 학습 작업 중지 또는 취소.

• 모델 성능을 검토하고 학습 실행을 비교하세요. 참고 학습된 모델 평가.

• 다음을 사용해 모델을 배포하세요 Serverless Cloud API.