lightly-ai/lightly
A python library for self-supervised learning on images.
해결하는 문제
자기 지도 학습(SSL)은 모델이 레이블이 없는 데이터로부터 학습할 수 있도록 하여 수동 데이터 어노테이션에 대한 과도한 의존도를 줄여줍니다. 이 프레임워크는 처음부터 시작하지 않고도 복잡한 SSL 알고리즘을 구현하는 데 필요한 빌딩 블록을 제공합니다.
작동 방식
Lightly는 손실 함수 및 모델 헤드와 같은 저수준 구성 요소를 노출하는 모듈식 프레임워크를 제공합니다. PyTorch와 함께 사용하도록 설계되었으며 PyTorch Lightning을 통한 분산 학습을 지원합니다. 사용자는 다양한 백본을 실험하고 PyTorch와 유사한 인터페이스를 통해 최첨단 SSL 방법을 구현할 수 있습니다.
대상 사용자
고급 자기 지도 학습 기술을 사용하여 레이블이 없는 데이터 세트에서 모델을 사전 학습해야 하는 컴퓨터 비전 연구자 및 머신러닝 엔지니어.