Repolode
제출하기
lightly-ai

lightly-ai/lightly

레이블 없이 비전 모델을 학습할 수 있도록 PyTorch 구성 요소를 제공하는 이미지 자기지도 학습용 Python 라이브러리입니다.

GitHub에서 보기웹사이트
스타
3,830
포크
379
열린 이슈
119
마지막 커밋
어제

어떤 프로젝트인가요?

Lightly SSL은 레이블이 없는 데이터로 이미지 모델을 학습하는 기법인 자기지도 학습을 위한 컴퓨터 비전 프레임워크입니다. ImageNet 스타일의 사전 학습 가중치에만 의존하지 않고 자체 이미지 컬렉션으로 백본을 사전 학습하려는 개발자를 대상으로 합니다.

이 라이브러리는 PyTorch와 비슷한 스타일로 작성되었으며, 손실 함수, projection 및 prediction head, 데이터 변환과 같은 저수준 구성 요소를 제공합니다. MoCo, SimCLR, BYOL, DINO, DINOv2, MAE, VICReg 등 여러 발표된 방법의 구현과 예제를 포함하며, PyTorch 및 PyTorch Lightning 예제도 제공합니다. 사용자 지정 백본 모델을 지원하고, PyTorch Lightning을 통한 분산 멀티 GPU 학습도 가능합니다.

pip으로 설치할 수 있으며 MIT 라이선스를 따릅니다. 유지 관리자는 데이터 큐레이션 및 라벨링을 위한 별도의 상용 제품도 제공하지만, 해당 제품은 이 패키지에 포함되지 않습니다.

핵심 기능

  • 01손실 함수와 모델 head 등의 모듈형 구성 요소
  • 02SimCLR, BYOL, DINO, MAE 등을 포함한 다양한 자기지도 학습 방법 구현
  • 03사용자 지정 백본 모델 지원
  • 04PyTorch 및 PyTorch Lightning 예제
  • 05손실 취합 및 BatchNorm 동기화를 지원하는 분산 학습

활용 사례

  • 도메인에 특화된 레이블 없는 이미지 데이터셋으로 비전 백본 사전 학습
  • 동일한 데이터에서 여러 자기지도 학습 방법 비교
  • 교체 가능한 손실 함수와 head를 활용해 사용자 지정 SSL 연구 실험 구축
  • PyTorch Lightning으로 여러 GPU에서 학습

사용 전 알아둘 점

  • Docker 지원과 단일 명령어로 사용할 수 있는 사전 학습 모델은 상용 버전에서만 제공됩니다.

포함된 컬렉션

컴퓨터 비전 라이브러리 및 도구

토픽

#computer-vision#self-supervised-learning#pytorch#deep-learning#contrastive-learning#embeddings#contributions-welcome#hacktoberfest#machine-learning

자주 묻는 질문

lightly은(는) 무엇인가요?
레이블 없이 비전 모델을 학습할 수 있도록 PyTorch 구성 요소를 제공하는 이미지 자기지도 학습용 Python 라이브러리입니다. Lightly SSL은 레이블이 없는 데이터로 이미지 모델을 학습하는 기법인 자기지도 학습을 위한 컴퓨터 비전 프레임워크입니다. ImageNet 스타일의 사전 학습 가중치에만 의존하지 않고 자체 이미지 컬렉션으로 백본을 사전 학습하려는 개발자를 대상으로 합니다.
lightly은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
lightly은(는) MIT 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
lightly은(는) 아직 유지보수되고 있나요?
활발히 유지보수 중. lightly의 마지막 커밋은 2026년 11월 6일입니다.
lightly은(는) 어디에 사용할 수 있나요?
도메인에 특화된 레이블 없는 이미지 데이터셋으로 비전 백본 사전 학습 동일한 데이터에서 여러 자기지도 학습 방법 비교 교체 가능한 손실 함수와 head를 활용해 사용자 지정 SSL 연구 실험 구축 PyTorch Lightning으로 여러 GPU에서 학습
lightly은(는) 어디서 받을 수 있나요?
소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/lightly-ai/lightly

요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.

관련 프로젝트