YOLO Vision 2024에서 소개된 Lightning AI가 더 빠른 모델 학습, 배포 및 협업으로 확장 가능한 비전 AI 개발을 간소화하는 방법을 살펴보세요.
%20(1).png)
숙련된 AI 개발자이든 이제 막 비전 AI를 탐구하기 시작한 사람이든, Ultralytrics ( YOLO11 )와 같은 컴퓨터 비전 모델을 가지고 놀고 실험할 수 있는 안정적인 환경을 갖추는 것이 중요합니다. 환경이란 AI 모델을 효율적으로 설계, 테스트 및 배포하는 데 필요한 도구, 리소스 및 인프라를 말합니다.
여러 온라인 플랫폼에서 다양한 AI 도구를 제공하지만, 데이터 준비부터 모델 배포까지 전체 AI 라이프사이클을 위한 통합 환경을 제공하는 플랫폼은 많지 않습니다. 바로 이 점에서 AI 개발을 위한 올인원 플랫폼인 Lightning AI가 데이터 준비부터 배포까지 프로세스를 간소화합니다.
AI와 컴퓨터 비전의 발전에 초점을 맞춘 Ultralytics 에서 주최하는 연례 하이브리드 행사인 YOLO 비전 2024(YV24)에서 AI 개발을 더 쉽게 만드는 것의 관련성이 소개되었습니다. Lightning AI의 CTO인 루카 안티가(Luca Antiga)는 기조연설에서'Lightning 스튜디오에서 YOLO '라는 제목의 강연을 통해 복잡한 기술 없이 빠르고 원활하게 Ultralytics YOLO 모델을 훈련하는 방법을 자세히 설명했습니다.
이 글에서는 실제 컴퓨터 비전 애플리케이션부터 Lightning AI를 사용한 훈련 및 배포에 대한 라이브 데모( Ultralytics YOLO 모델)에 이르기까지 Luca의 강연에서 얻은 핵심 내용을 자세히 살펴보겠습니다. 시작해 보겠습니다!
루카는 다양한 산업 분야에서 YOLO 모델의 영향력에 대한 자신의 생각과 감사를 공유하며 기조연설을 시작했습니다. 그는 YOLO 모델이 제조 및 농업과 같은 분야에 어떻게 적용될 수 있는지를 강조했습니다. 그는 " YOLO 이 실제적이고 실질적인 문제를 해결해야 하는 건축가 커뮤니티에 미친 영향에 감사한다"며 "이는 저와 매우 밀접한 관련이 있다"고 말했습니다.
그는 AI 훈련에 대한 관심이 높아지는 것과 연결하여 누구나 더 빠르고, 더 간단하고, 더 쉽게 AI 모델을 개발할 수 있도록 설계된 플랫폼인 Lightning AI를 소개했습니다. 특히 AI의 반복적인 발전을 지원하여 개발자가 모델을 개선하고 개선할 수 있도록 돕는 데 유용합니다.
그는 또한 Lightning AI가 AI 모델 학습 과정을 간소화하는 프레임워크인 PyTorch Lightning과 유사하다고 지적했습니다. 하지만 Lightning AI는 AI 모델 학습뿐만 아니라 전체 AI 개발 프로세스를 위한 광범위한 도구와 기능을 제공하는 보다 포괄적인 플랫폼이라는 점에서 차이가 있습니다.
AI 모델을 설계, 학습 및 배포할 수 있는 직관적인 작업 공간을 제공하여 전체 워크플로우를 원활하고 효율적으로 만들어주는 Lightning AI의 핵심 구성 요소는 Lightning Studios입니다. Lightning Studios는 클라우드에서 실행되는 재현 가능한 AI 개발 환경이라고 생각하면 됩니다. 예를 들어, 다른 개발자와 복제 및 공유할 수 있는 Jupyter 노트북과 유사한 환경을 제공하여 협업을 개선하는 데 도움이 됩니다.
이어서 루카는 "환경을 복제하는 것은 더 이상 문제가 되지 않습니다."라며 라이트닝 스튜디오의 장점에 대해 자세히 설명했습니다. CPU [중앙 처리 장치] 머신에서 GPU [그래픽 처리 장치] 머신으로 변경하거나 수천 대의 머신에서 교육을 시작해야 하는 경우에도 환경이 영구적으로 유지됩니다."라고 설명합니다.
다음으로 루카는 Lightning 스튜디오를 얼마나 빠르게 시작할 수 있는지 시연했습니다. 몇 번의 클릭만으로 새 스튜디오를 열고 Jupyter Notebook 및 VS Code와 같은 도구와 환경에 액세스하여 코딩할 수 있는 모든 설정과 준비를 마칠 수 있습니다. 그는 다른 기기 간에 전환하는 것이 얼마나 쉬운지 보여주었습니다. 작업 중인 작업에 더 강력한 성능이 필요한 경우 CPU 에서 더 강력한 GPU 으로 쉽게 전환할 수 있습니다. GPU 은 사용 중일 때만 활성 상태로 유지되며, 그 외에는 절전 모드로 전환되어 크레딧이 절약됩니다.
루카는 스튜디오 템플릿 사용의 장점에 대해서도 언급했습니다. 커뮤니티에서 미리 만든 AI 코딩 환경으로, 아무것도 설정할 필요 없이 바로 사용할 수 있습니다. AI 프로젝트를 위한 환경을 설정하는 데는 많은 시간이 소요될 수 있는데, 스튜디오 템플릿은 생산성을 높이는 데 도움이 될 수 있습니다. 이러한 환경에는 설치된 종속성, 모델 가중치, 데이터, 코드 등 AI 프로젝트에 필요한 모든 것이 미리 로드되어 있습니다.
그런 다음 Luca는 라이브 데모로 넘어가서 Lightning Studio를 사용하여 Ultralytics YOLO 모델을 훈련하는 방법을 강조했습니다. 그는 이미 모든 종속성이 설치된 스튜디오 템플릿을 열고 4개의 GPU가 장착된 머신을 가동하여 훈련 프로세스의 속도를 높였습니다. 데이터와 관련해서는 머신에 직접 데이터를 저장하거나 클라우드에서 스트리밍하여 훈련 프로세스를 더 빠르고 효율적으로 만들 수 있다고 설명했습니다.
몇 초 만에 머신이 준비되었고 루카는 빠르게 교육 세션을 시작했습니다. 데모 도중 사소한 문제로 인해 머신이 예기치 않게 멈췄지만 Lightning 스튜디오는 중단된 지점부터 원활하게 다시 시작하여 진행 상황을 잃지 않았습니다. 루카는 이러한 안정성이 예기치 않은 중단에도 원활한 워크플로를 지원하는 방법에 대해 설명했습니다.
데모를 이어서 그는 머신러닝 메트릭을 실시간으로 시각화하는 도구인 TensorBoard를 사용해 훈련 진행 상황을 모니터링하는 것이 얼마나 쉬운지 보여줬습니다. Lightning Studio는 별도의 설정 없이도 같은 작업 공간에 있는 팀원들이 TensorBoard 보기에 액세스할 수 있는 URL을 자동으로 생성하여 이 작업을 더욱 간단하게 만들어 줍니다. 이를 통해 협업을 간소화하고 모든 사람이 같은 정보를 공유할 수 있습니다.
데모가 끝난 후, 루카는 강연의 초점을 최근 Lightning AI에서 시작한 새로운 프로젝트인 LitServe로 옮겼습니다. LitServe는 학습된 모델을 다른 사람이 사용할 수 있는 확장 가능한 서비스로 전환하는 프로세스를 간소화하여 복잡한 배포 파이프라인이 필요하지 않습니다. 모델 패키징부터 배포까지 모든 작업을 최소한의 노력으로 처리할 수 있도록 설계되었습니다.
이를 실시간으로 보여주기 위해 루카는 미리 훈련된 모델을 사용하여 청중에게 간단한 데모를 보여주었습니다. Ultralytics YOLOv8 모델을 사용해 간단한 데모를 보여주었습니다. 그는 들어오는 요청을 처리하고 몇 초 만에 이미지 예측을 반환하는 간단한 API를 만들 수 있었습니다. 즉, 누구나 이 API에 이미지를 핑하여 물체 감지와 같은 컴퓨터 비전 작업에 대한 결과를 거의 즉시 받을 수 있습니다. 그 이면에는 Ultralytics YOLOv8 모델이 서비스 형태로 배포되어 요청을 효율적으로 처리하고 이미지를 처리하며 최소한의 지연 시간으로 예측을 제공합니다.
그는 피자 이미지에 대한 추론을 실행했고 Ultralytics YOLOv8 피자, 숟가락, 식탁과 같은 물체를 성공적으로 식별했습니다. 그는 첫 번째 요청은 '콜드 스타트' 때문에 시간이 조금 더 걸리지만, 시스템이 예열되면 후속 요청은 훨씬 더 빨라진다고 설명했습니다.
그런 다음 루카는 '이 모델을 외부에 노출하려면 어떻게 해야 하나요?"라고 질문했습니다. 그는 API 빌더 플러그인을 통해 모델을 실제 프로덕션에 사용할 수 있는 서비스로 간단하게 전환하는 방법을 설명했습니다. 사용자 지정 도메인, 보안 강화, 원활한 통합과 같은 기능을 통해 누구나 쉽게 모델에 액세스할 수 있도록 만들 수 있습니다.
루카는 강연을 마무리하며 AI 개발을 위한 Lightning Studio의 확장성과 유연성에 대해 설명했습니다. 그는 플랫폼이 여러 머신에서 모델을 훈련하고 최대 10,000개의 노드로 확장할 수 있으며, 중단 후 자동으로 재개되는 내결함성 훈련을 통해 어떻게 모델을 훈련할 수 있는지에 대해 언급했습니다.
예를 들어, 하드웨어 문제나 서버 재부팅으로 인해 GPU 클러스터의 트레이닝 작업이 중단된 경우, Lightning Studios는 프로세스가 중단된 지점부터 정확히 다시 시작되도록 합니다. 따라서 이미지넷이나 코코와 같은 대규모 데이터 세트에 대한 딥러닝 모델 훈련과 같은 대규모 AI 프로젝트에 이상적입니다.
루카가 소개한 라이트닝 스튜디오의 다른 주요 장점은 다음과 같습니다:
루카는 YV24의 기조연설에서 AI가 Ultralytics YOLO 모델 및 Lightning AI와 같은 도구와 결합하여 실제 문제를 해결하는 방식을 어떻게 변화시키고 있는지에 대해 강조했습니다. 이러한 도구를 사용하면 개발자가 다양한 산업 분야의 특정 문제를 해결하도록 설계된 모델을 더 쉽게 훈련하고 배포할 수 있습니다.
그는 라이트닝 스튜디오가 어떻게 전체 개발 프로세스를 더 빠르고 더 쉽게 만들어 개발자가 강력한 솔루션을 쉽게 만들 수 있도록 지원하는지 설명했습니다. Lightning AI와 같은 최첨단 플랫폼의 핵심인 컴퓨터 비전 모델은 AI 솔루션이 문제를 처리하는 방식을 혁신하고 있습니다. 특히 최신 Ultralytics YOLO11 모델을 통해 개발자는 의미 있는 영향력을 발휘하는 솔루션을 구축할 수 있습니다.
커뮤니티에 가입하여 AI와 그 실제 활용에 대한 최신 소식을 받아보세요. GitHub 리포지토리에서 자율 주행 자동차 의 AI와 의료 분야의 컴퓨터 비전과 같은 분야의 혁신을 살펴보세요.
