데이터를 다양화하기 위한 Albumentations 증강 사용

컴퓨터 비전 솔루션을 구축할 때, 비전 AI 모델 학습을 위해 다양한 이미지 세트를 수집하는 것은 매우 중요한 과정입니다. 이는 종종 많은 시간과 비용을 필요로 하며, 때로는 수집된 이미지가 모델이 효과적으로 학습하기에 충분히 다양하지 않을 수도 있습니다.

예를 들어, Ultralytics YOLO11과 같은 컴퓨터 비전 모델은 다양한 컴퓨터 비전 작업과 관련된 이미지 데이터셋으로 커스텀 학습될 수 있습니다. 다양한 데이터는 모델의 일반화 성능을 높여주며, 실제 환경의 다양한 상황에서 객체와 패턴을 인식할 수 있게 해주기 때문에 매우 중요합니다.

데이터 다양성 부족으로 어려움을 겪고 있다면, 이미지 데이터 증강 기법이 훌륭한 해결책이 될 수 있습니다. 회전, 뒤집기, 밝기 조절과 같은 방법들은 데이터셋의 다양성을 높여 모델이 더 광범위한 조건을 처리할 수 있도록 성능을 향상시킵니다.

이러한 이유로 Ultralytics는 이미지 데이터 증강 통합 기능을 지원합니다. 다양한 변환을 제공하는 대중적인 도구인 Albumentations를 사용하면 다채로운 시각적 데이터를 생성할 수 있습니다. 이 통합 기능은 학습 이미지를 자동으로 증강하여 YOLO11 학습 과정을 간소화하고, 결과적으로 모델 성능을 개선합니다.

본 글에서는 Albumentations 통합 기능을 사용하는 방법과 그 이점, 그리고 모델 학습에 미치는 영향에 대해 알아봅니다.

Link to this sectionAlbumentations이란 무엇인가요?#

컴퓨터 비전 모델은 다양한 환경에서 객체를 인식하기 위해 광범위하고 고품질인 이미지 세트로부터 학습할 수 있습니다. 실제 환경에서 대규모 데이터셋을 수집하는 것은 느리고 비용이 많이 들며 비효율적일 수 있습니다. 이 작업을 효율화하기 위해 이미지 데이터 증강을 사용하여 기존 이미지의 새로운 변형을 생성하면, 더 많은 데이터를 수집하지 않고도 모델이 다양한 상황을 학습하도록 도울 수 있습니다.

구체적으로는 2018년에 효율적인 이미지 데이터 증강을 위해 도입된 오픈 소스 라이브러리인 Albumentations을 활용할 수 있습니다. 회전이나 뒤집기와 같은 단순한 기하학적 변경부터 밝기, 대비, 노이즈 추가와 같은 복잡한 조정에 이르기까지 다양한 작업을 지원합니다.

그림 1. 다양한 유형의 이미지 데이터 증강 예시.

Link to this sectionAlbumentations의 주요 특징#

Albumentations은 고성능으로 잘 알려져 있으며, 이는 이미지를 빠르고 효율적으로 처리할 수 있음을 의미합니다. OpenCV 및 NumPy와 같이 최적화된 라이브러리를 기반으로 구축되어 대규모 데이터셋을 최소한의 처리 시간으로 다룰 수 있으므로, 모델 학습 중 빠른 데이터 증강에 이상적입니다.

Albumentations의 다른 주요 특징은 다음과 같습니다:

• 다양한 변환: Albumentations은 70가지 이상의 증강 유형을 제공합니다. 이러한 변화는 모델이 조명, 각도 또는 배경 변화에도 불구하고 객체를 감지하는 방법을 학습하도록 돕습니다.
• 속도 최적화: SIMD(Single Instruction, Multiple Data)와 같은 고급 최적화 기술을 사용하여 여러 데이터 포인트를 동시에 처리함으로써 이미지 증강 속도를 높이고 대규모 데이터셋을 효율적으로 처리합니다.
• 세 가지 수준의 증강: 세 가지 방식으로 데이터를 강화합니다. 예를 들어, 픽셀 수준 증강은 객체를 변경하지 않고 밝기와 색상을 조정합니다. 반면 공간 수준 증강은 핵심 세부 정보를 유지하면서 객체 위치를 수정하며, 혼합 수준 증강은 여러 이미지의 일부를 결합하여 새로운 샘플을 생성합니다.

Link to this section왜 Albumentations 통합 기능을 사용해야 할까요?#

데이터셋에 증강을 적용하는 방법은 많으며, OpenCV와 같은 도구를 사용하여 직접 만들 수도 있는데 왜 Albumentations 같은 라이브러리를 지원하는 통합 기능을 선택해야 할지 궁금할 수 있습니다.

OpenCV와 같은 도구로 직접 증강을 만드는 것은 시간이 많이 걸리고 전문 지식이 필요할 수 있습니다. 또한 최상의 결과를 얻기 위해 변환을 세밀하게 조정하는 것도 까다로울 수 있습니다. Albumentations 통합은 이 과정을 더 쉽게 만듭니다. 즉시 사용 가능한 다양한 변환을 제공하여 데이터셋을 준비할 때 시간과 노력을 절약할 수 있습니다.

Albumentations 통합 기능을 선택해야 하는 또 다른 이유는 Ultralytics 모델 학습 파이프라인과 원활하게 작동하기 때문입니다. 학습 중에 증강이 자동으로 적용되므로 YOLO11 커스텀 학습이 훨씬 쉬워집니다. 데이터 준비 작업을 직접 처리하는 대신 모델 개선에 더 집중할 수 있도록 과정을 간소화합니다.

Link to this sectionAlbumentations 통합 시작하기#

흥미롭게도, Albumentations 통합 기능을 사용하여 YOLO11을 학습시키는 것은 생각보다 훨씬 간단합니다. 올바른 라이브러리가 설정되면 통합 기능이 학습 중에 이미지 데이터 증강을 자동으로 적용합니다. 동일한 데이터셋을 사용하여 모델이 다양한 이미지 변형을 학습하도록 돕습니다.

다음으로, YOLO11 커스텀 학습 시 Albumentations 통합 기능을 설치하고 사용하는 방법을 살펴보겠습니다.

Link to this sectionUltralytics Python 패키지 및 Albumentations 설치#

증강을 적용하기 전에 Ultralytics Python 패키지와 Albumentations가 모두 설치되어 있어야 합니다. 통합 기능은 두 라이브러리가 기본적으로 원활하게 함께 작동하도록 구축되었으므로 복잡한 구성에 대해 걱정할 필요가 없습니다.

전체 설치 과정은 아래 이미지와 같이 Python 라이브러리 설치를 위한 패키지 관리 도구인 pip 명령 하나로 단 몇 분 만에 완료할 수 있습니다.

그림 2. Ultralytics 및 Albumentations 설치.

Albumentations이 설치되면 Ultralytics 모델 학습 모드에서 학습 중에 자동으로 이미지 증강이 적용됩니다. Albumentations이 설치되어 있지 않으면 이러한 증강은 적용되지 않습니다. 자세한 내용은 공식 Ultralytics 문서를 참조하십시오.

Link to this sectionAlbumentations 통합을 통한 YOLO11 학습#

Albumentations 통합 기능의 내부에서 어떤 일이 일어나는지 더 자세히 알아보겠습니다.

YOLO11 학습 중에 적용되는 증강을 자세히 살펴보겠습니다:

• Blur(흐림 효과): 이 변환은 이미지에 약간의 흐림 효과를 추가합니다. 모델이 초점이 맞지 않는 객체도 감지할 수 있도록 돕습니다.
• Median blur(중앙값 흐림 효과): 이미지의 객체 가장자리를 보존하면서 무작위 노이즈를 줄입니다. 이를 통해 모델이 복잡한 환경에서 객체를 더 쉽게 감지할 수 있습니다.
• Grayscale(회색조): 이미지를 흑백으로 변환하여 모델이 색상 대신 형태와 질감에 집중하도록 도울 수 있습니다.
• CLAHE(대비 제한 적응형 히스토그램 평활화): 이 증강은 이미지의 대비를 높이며, 특히 조명이 어둡거나 가시성이 낮은 조건에서 잘 보이지 않는 영역의 대비를 개선합니다. 이렇게 하면 해당 영역의 객체가 더 명확해져 모델이 식별하기 쉬워집니다.

그림 3. 고양이 이미지에 적용된 회색조 증강의 예시.

Link to this sectionYOLO11 및 Albumentations 통합의 응용 분야#

특정 애플리케이션을 위해 YOLO11을 커스텀 학습하는 경우, Albumentations 통합을 사용하여 다양한 조건에 적응함으로써 모델 성능을 향상시킬 수 있습니다. 실제 응용 분야와 이 통합 기능이 해결할 수 있는 과제에 대해 논의해 보겠습니다.

Link to this section의료 영상 개선#

의료 분야의 비전 AI는 의사들이 의료 영상을 더 정확하게 분석하여 진단을 돕고 환자 치료를 개선하도록 지원합니다. 실제로 의료 기관의 약 5분의 1이 이미 의료 솔루션에 AI 모델을 도입하고 있습니다.

그러나 이러한 컴퓨터 비전 솔루션을 구축하는 데는 어려움이 따릅니다. 의료 스캔 영상은 장비, 설정, 심지어 기술자의 경험과 같은 요인에 따라 병원마다 크게 다를 수 있습니다. 밝기, 대비, 노출의 차이는 비전 AI 모델의 일관성과 정확성에 영향을 미칠 수 있으며, 다양한 환경에서 안정적으로 작동하기 어렵게 만듭니다.

이때 Albumentations와 같은 도구의 통합이 필수적입니다. 동일한 스캔 영상의 다양한 증강 버전을 생성함으로써 Albumentations은 모델이 여러 이미지 품질을 학습할 수 있게 합니다. 이는 모델을 더 강력하게 만들어 고품질 이미지와 저품질 이미지 모두에서 질병을 정확하게 감지할 수 있도록 돕습니다.

그림 4. 증강된 X-레이 이미지.

Link to this section보안 및 감시 강화#

비전 AI의 또 다른 흥미로운 응용 분야는 보안 및 감시입니다. 실시간 객체 감지는 보안 팀이 잠재적인 위협을 신속하게 식별하도록 도울 수 있습니다.

이 애플리케이션과 관련된 주요 우려 사항은 보안 카메라가 하루 종일 다양한 조명 조건에서 영상을 촬영한다는 점이며, 이러한 조건은 모델이 이미지를 이해하는 방식에 큰 영향을 줄 수 있습니다. 저조도 환경, 눈부심 또는 낮은 가시성과 같은 요소들은 컴퓨터 비전 모델이 객체를 감지하거나 잠재적인 위협을 일관되게 인식하기 어렵게 만듭니다.

Albumentations 통합은 다양한 조명 조건을 모방하도록 변환을 적용함으로써 도움을 줍니다. 이를 통해 모델은 밝은 환경과 저조도 환경 모두에서 객체를 감지하는 법을 학습하며, 더 신뢰할 수 있게 되고 까다로운 조건에서도 대응 시간을 개선합니다.

Link to this section소매 워크플로우 및 고객 경험 재정의#

슈퍼마켓 통로에 쏟아진 물건, 매장을 뛰어다니는 개, 제품 진열대를 넘어뜨리는 아이는 소매업 분야의 비전 AI 환경에서 발생할 수 있는 에지 케이스(Edge case) 사례들입니다. 컴퓨터 비전은 고객 행동 추적, 유동 인구 모니터링, 선반 위 제품 식별 등을 통해 고객 경험을 개선하는 데 점점 더 많이 사용되고 있습니다. 그러나 이러한 실제 상황은 AI 시스템이 이해하고 정확하게 처리하기 어려울 수 있습니다.

모든 시나리오를 컴퓨터 비전 데이터셋에 반영할 수는 없지만, Albumentations 통합은 예상치 못한 조명, 독특한 각도 또는 장애물과 같은 많은 가능한 상황을 다룰 수 있도록 데이터를 증강하여 도움을 줍니다. 이는 컴퓨터 비전 모델이 다양한 조건에 적응하도록 하여 에지 케이스를 처리하고 역동적인 소매 환경에서 정확한 예측을 수행하는 능력을 향상시킵니다.

Link to this section핵심 요약#

모델 학습을 위해 다양한 실제 데이터를 수집하는 것은 복잡할 수 있지만, Albumentations은 모델이 다른 조건에 적응하도록 돕는 이미지 변형을 생성하여 이를 더 쉽게 만듭니다.

Ultralytics에서 지원하는 Albumentations 통합은 YOLO11 커스텀 학습 중에 이러한 증강을 적용하는 과정을 간소화합니다. 이는 더 정확하고 신뢰할 수 있는 비전 AI 모델을 생성함으로써 광범위한 산업에 혜택을 주는 더 나은 데이터셋 품질로 이어집니다.

우리 커뮤니티에 가입하고 GitHub 저장소를 탐색하여 AI에 대해 더 자세히 알아보고, 라이선스 옵션을 확인하여 비전 AI 프로젝트를 시작해 보십시오. 제조업에서의 AI나 자율주행에서의 컴퓨터 비전과 같은 혁신에 관심이 있으신가요? 당사의 솔루션 페이지를 방문하여 더 많은 정보를 확인하십시오.