Google Colab에서 Ultralytics YOLO11로 이미지 세그멘테이션하기

Ultralytics YOLO11 및 후속 모델인 Ultralytics YOLO26과 같은 Ultralytics YOLO 모델은 객체 감지, 인스턴스 분할, 이미지 분류 등의 다양한 컴퓨터 비전 작업을 지원합니다. 이러한 작업은 각각 인간 시각의 특정 측면을 모방하여 기계가 주변 세계를 보고 해석할 수 있도록 합니다.

예를 들어 미술 수업에서 학생이 연필을 들고 그림 속 객체의 윤곽을 그리는 상황을 생각해 보세요. 이 과정에서 학생의 뇌는 객체를 배경 및 다른 요소와 구분하는 분할 작업을 수행합니다. 이미지 분할은 인공지능(AI)을 사용해 시각 데이터를 의미 있는 부분으로 나누어 기계가 이해하도록 한다는 점에서 비슷한 목표를 달성합니다. 이 기술은 여러 산업 분야의 다양한 애플리케이션에 활용될 수 있습니다.

그림 1. 이미지에서 객체를 분할하는 데 사용되는 Ultralytics YOLO11.

실용적인 예로 자동차 부품 분할이 있습니다. 차량의 특정 부품을 식별하고 분류하면 자동차 제조, 수리, 전자상거래 카탈로그 작성과 같은 산업 분야의 프로세스를 간소화할 수 있습니다.

이 글에서는 Ultralytics YOLO11, Google Colab, Roboflow Carparts Segmentation 데이터셋을 사용하여 자동차 부품을 정확하게 식별하고 분할할 수 있는 솔루션을 구축하는 방법을 살펴보겠습니다.

Ultralytics YOLO11은 사용하기 쉽습니다#

Ultralytics YOLO11은 80개의 서로 다른 객체 클래스를 포함하는 COCO 데이터셋으로 학습된 사전 학습 모델로 제공됩니다. 하지만 자동차 부품 분할과 같은 특정 애플리케이션에서는 데이터셋과 사용 사례에 더 잘 맞도록 모델을 맞춤 학습할 수 있습니다. 이러한 유연성 덕분에 YOLO11은 범용 작업과 고도로 전문화된 작업 모두에서 뛰어난 성능을 발휘합니다.

사용자 지정 학습은 사전 학습된 Ultralytics YOLO11 모델을 새로운 데이터셋으로 미세 조정하는 과정입니다. 작업에 특화된 라벨링 예제를 제공하면 모델이 프로젝트에만 해당하는 객체를 인식하고 분할하도록 학습됩니다. 사용자 지정 학습은 일반적인 사전 학습 가중치에 의존하는 것보다 높은 정확도와 관련성을 제공하는 경우가 많습니다.

맞춤 학습을 위한 Ultralytics YOLO11 설정은 간단합니다. 최소한의 설정만으로 모델과 데이터셋을 불러오고 학습을 시작할 수 있으며, 학습 중 손실 및 정확도와 같은 지표를 모니터링할 수 있습니다. YOLO11에는 검증 및 평가를 위한 도구도 내장되어 있어 모델의 성능을 더 쉽게 평가할 수 있습니다.

Google Colab에서 Ultralytics YOLO11 실행하기#

Ultralytics YOLO11을 맞춤 학습할 때는 환경을 설정하는 방법이 몇 가지 있습니다. 가장 쉽게 접근할 수 있고 편리한 선택지 중 하나는 Google Colab입니다. YOLO11 학습에 Google Colab을 사용하면 다음과 같은 이점이 있습니다.

• 리소스 무료 이용: Google Colab은 GPU(그래픽 처리 장치)와 TPU(텐서 처리 장치)를 제공하므로, 고가의 하드웨어 없이 Ultralytics YOLO11을 학습할 수 있습니다.
• 협업 환경: Google Colab을 사용하면 노트북을 공유하고, Google Drive에 작업을 저장하며, 손쉬운 협업과 버전 추적을 통해 팀워크를 간소화할 수 있습니다.
• 라이브러리 사전 설치: PyTorch와 TensorFlow 같은 도구가 사전 설치되어 있어 Google Colab은 설정 과정을 간소화하고 빠르게 시작할 수 있도록 도와줍니다.
• 클라우드 통합: Google Drive, GitHub 또는 기타 클라우드 소스에서 데이터셋을 쉽게 불러올 수 있어 데이터 준비와 저장을 간소화합니다.

그림 2. Google Colab Ultralytics YOLO11 노트북.

Ultralytics는 YOLO11 학습 전용으로 사전 구성된 Google Colab 노트북도 제공합니다. 이 노트북에는 모델 학습부터 성능 평가까지 필요한 모든 기능이 포함되어 있어 과정을 간단하고 쉽게 따라갈 수 있습니다. 복잡한 설정 단계에 신경 쓰지 않고 특정 요구 사항에 맞게 모델을 세부 조정하는 데 집중할 수 있어 좋은 출발점이 됩니다.

Roboflow Carparts Segmentation 데이터셋 개요#

학습 환경을 정했다면 다음 단계는 데이터를 수집하거나 자동차 부품 분할에 적합한 데이터셋을 선택하는 것입니다. Roboflow Universe에서 이용할 수 있는 Roboflow Carparts Segmentation 데이터셋은 컴퓨터 비전 모델 구축, 학습, 배포 도구를 제공하는 플랫폼인 Roboflow에서 관리합니다. 이 데이터셋에는 범퍼, 도어, 미러, 휠 등 자동차 부품에 대한 고품질 주석이 포함된 학습 이미지 3,156개, 검증 이미지 401개, 테스트 이미지 276개가 있습니다.

일반적으로 Roboflow Universe에서 데이터셋을 다운로드한 다음 Google Colab에서 학습할 수 있도록 수동으로 설정해야 합니다. 하지만 Ultralytics Python 패키지는 내장된 데이터셋 구성을 제공하여 이 과정을 간소화합니다.

그림 3. 자동차 부품 분할 데이터셋 예시.

Ultralytics에서는 데이터셋 경로, 클래스 레이블, 기타 학습 매개변수가 포함된 사전 구성 YAML 파일을 통해 데이터셋을 바로 사용할 수 있습니다. 이 파일이 설정을 대신 처리하므로 데이터셋을 빠르게 불러와 바로 모델 학습을 시작할 수 있습니다. 또한 데이터셋은 학습, 검증, 테스트 세트로 구분되어 있어 진행 상황을 모니터링하고 성능을 평가하기 쉽습니다.

Roboflow Carparts Segmentation Dataset과 Ultralytics YOLO11이 제공하는 도구를 사용하면 Google Colab과 같은 플랫폼에서 분할 모델을 효율적으로 구축하는 간소화된 워크플로를 이용할 수 있습니다. 이 방식은 설정 시간을 줄여 실제 애플리케이션에 맞게 모델을 개선하는 데 집중할 수 있도록 합니다.

자동차 부품 분할의 실제 활용 사례#

자동차 부품 분할은 여러 산업 분야에서 다양하게 활용됩니다. 예를 들어 정비소에서는 손상된 부품을 신속하게 식별하고 분류하여 수리 과정을 더 빠르고 효율적으로 진행할 수 있습니다. 마찬가지로 보험 업계에서는 손상된 차량 이미지를 분석해 영향을 받은 부품을 식별함으로써 분할 모델로 보험금 청구 평가를 자동화할 수 있습니다. 이를 통해 보험금 청구 절차가 빨라지고 오류가 줄어들며 보험사와 고객 모두 시간을 절약할 수 있습니다.

그림 4. YOLO를 사용한 자동차 부품 분할.

제조 분야에서 분할은 자동차 부품의 결함을 검사하고 일관성을 보장하며 폐기물을 줄여 품질 관리를 지원합니다. 이러한 활용 사례는 자동차 부품 분할이 프로세스를 더 안전하고 빠르며 정확하게 만들어 산업을 혁신할 수 있음을 보여줍니다.

단계별 가이드: Google Colab에서 Ultralytics YOLO11 사용하기#

세부 사항을 모두 살펴보았으니 이제 전체 과정을 진행해 보겠습니다. 시작하려면 자동차 부품 분할을 위한 Ultralytics YOLO11 모델의 설정, 학습, 검증 과정을 안내하는 YouTube 동영상을 확인하세요.

관련 단계를 간단히 살펴보겠습니다.

• Google Colab 환경 설정: 모델 학습을 준비하기 위해 GPU 지원을 활성화하고 Ultralytics Python 패키지를 설치합니다.
• Ultralytics YOLO11 모델 로드: 시간을 절약하고 자동차 부품 분할에 학습된 특징을 재사용하려면 사전 학습된 YOLO11 분할 모델로 시작하세요.
• 데이터셋으로 모델 학습: 학습 중 “carparts-seg.yaml” 파일을 사용하면 Roboflow Carparts Segmentation 데이터셋을 자동으로 다운로드하고 구성하여 사용할 수 있습니다. 에포크, 이미지 크기, 배치 크기 등의 매개변수를 조정해 모델을 세부 조정합니다.
• 학습 진행 상황 모니터링: 분할 손실과 평균 정밀도(mAP) 같은 주요 성능 지표를 추적하여 모델이 예상대로 개선되는지 확인합니다.
• 모델 검증 및 배포: 검증 세트에서 학습된 모델을 테스트하여 정확도를 확인한 다음, 품질 관리나 보험 청구 처리와 같은 실제 애플리케이션에 사용할 수 있도록 내보내기를 수행하세요.

자동차 부품 분할에 Ultralytics YOLO11을 사용할 때의 이점#

Ultralytics YOLO11은 자동차 부품 분할을 위한 신뢰할 수 있고 효율적인 도구로, 다양한 실제 애플리케이션에 적합한 여러 이점을 제공합니다. 주요 이점은 다음과 같습니다.

• 속도와 효율성: Ultralytics YOLO11은 높은 정확도를 유지하면서 이미지를 빠르게 처리하므로 품질 관리 및 자율주행차와 같은 실시간 작업에 적합합니다.
• 높은 정확도: 모델은 단일 이미지 내 여러 객체를 탐지하고 분할하는 데 뛰어나며, 자동차 부품을 정확하게 식별합니다.
• 확장성: Ultralytics YOLO11은 대규모 데이터셋과 복잡한 분할 작업을 처리할 수 있어 산업 애플리케이션에 맞게 확장할 수 있습니다.
• 다양한 통합: Ultralytics는 Google Colab, Ultralytics HUB(현재 Ultralytics Platform) 및 기타 인기 도구와의 통합을 지원하여 개발자의 유연성과 접근성을 높입니다.

Google Colab에서 Ultralytics YOLO11을 사용할 때 유용한 팁#

Google Colab은 머신러닝 워크플로를 훨씬 쉽게 처리할 수 있도록 해 주지만, 처음 사용하는 경우 익숙해지는 데 시간이 조금 걸릴 수 있습니다. 클라우드 기반 설정, 런타임 설정, 세션 제한을 처음에는 다루기 어렵게 느낄 수 있지만, 몇 가지 팁을 활용하면 훨씬 수월하게 사용할 수 있습니다.

다음 사항을 염두에 두세요.

• 학습 속도를 높이려면 런타임 설정에서 GPU 가속을 활성화하는 것부터 시작하세요.
• Colab은 클라우드에서 실행되므로 데이터셋과 저장소 같은 리소스에 액세스할 수 있도록 안정적인 인터넷 연결을 유지하세요.
• Colab에서 파일과 데이터셋을 쉽게 불러오고 관리할 수 있도록 Google Drive 또는 GitHub에 정리해 두세요.
• Colab 무료 티어에서 메모리 제한이 발생하면 학습 중 이미지 크기나 배치 크기를 줄여 보세요.
• Colab 세션에는 시간 제한이 있으므로 진행 상황을 잃지 않도록 모델과 결과를 정기적으로 저장하세요.

Ultralytics YOLO11로 더 많은 성과를 달성하세요#

Ultralytics YOLO11은 Google Colab과 같은 플랫폼 및 Roboflow Carparts Segmentation 데이터셋과 결합하면 이미지 세그멘테이션을 간편하고 쉽게 이용할 수 있게 합니다. 직관적인 도구, 사전 학습된 모델, 간편한 설정을 갖춘 YOLO11을 사용하면 고급 컴퓨터 비전 작업을 손쉽게 시작할 수 있습니다.

자동차 안전을 개선하거나 제조를 최적화하거나 혁신적인 AI 애플리케이션을 구축하려는 경우, 이 조합은 성공에 필요한 도구를 제공합니다. Ultralytics YOLO11을 사용하면 실제 문제를 해결하는 실용적이고 효율적인 솔루션을 구축할 수 있습니다.

자세한 내용은 GitHub 저장소를 확인하고 커뮤니티에 참여해 보세요. 솔루션 페이지에서 자율주행차 및 농업 분야의 컴퓨터 비전 관련 AI 애플리케이션을 살펴보세요.