lightly-ai/lightly-train

해결하는 문제

LightlyTrain은 최첨단 컴퓨터 비전 모델을 쉽게 만들 수 있도록 설계된 프레임워크입니다. 레이블이 없는 데이터에서 기초 모델을 사전 학습하고, 객체 탐지 및 세그멘테이션과 같은 특정 작업에 맞게 미세 조정한 후, 최종적으로 엣지 장치에 최적화하는 모델 개발 전 과정의 관리 어려움을 해결합니다.

작동 방식

이 프레임워크는 다양한 컴퓨터 비전 워크플로우를 처리하기 위한 고수준 API를 제공합니다. 시각 기초 모델(DINOv2 및 DINOv3 등)에서의 사전 학습 및 지식 증류, 전문화된 작업을 위한 미세 조정을 지원합니다. 또한 ONNX 및 TensorRT와 같은 최적화 도구와 통합되어 임베디드 장치에서 더 빠른 추론을 가능하게 합니다.

대상 사용자

엣지 장치나 고유 데이터셋을 타겟으로 하는 고성능 컴퓨터 비전 모델을 구축, 학습, 배포해야 하는 머신러닝 엔지니어 및 개발자에게 적합합니다.

주요 특징

• 포괄적인 작업 지원: 객체 탐지, 인스턴스 세그멘테이션, 패노픽 세그멘테이션, 세그멘테이션, 단안 깊이 추정을 지원합니다.
• 최신 아키텍처: LTDETRv2, DINOv3, EoMT(CVPR 2025)와 같은 최신 모델을 통합합니다.
• 엣지 최적화: 저전력 임베디드 장치를 위한 FP16 정밀도에서 ONNX 및 TensorRT 내보내기를 기본 지원합니다.
• 유연한 백본: ViTs, 하이브리드, CNNs 등 다양한 백본과 호환되며, 지식 증류용 사용자 정의 교사 모델도 지원합니다.