tracel-ai/models

해결하는 문제

이 저장소는 Burn 딥러닝 프레임워크를 사용하여 구축된 딥러닝 모델 구현 및 예제를 중앙 집중식으로 모아놓은 곳입니다. 개발자는 다양한 모달리티에 걸쳐 기존 모델 아키텍처를 쉽게 찾고 통합할 수 있습니다.

작동 방식

이 프로젝트는 공식 구현과 커뮤니티 기여 모델 모두를 위한 허브 역할을 합니다. 언어 모델링, 이미지 분류, 음성 인식과 같은 기능별로 모델을 정리하고 코드가 호스팅되는 특정 저장소에 대한 링크를 제공합니다.

대상 사용자

Burn 프레임워크를 사용하며 프로젝트에 즉시 사용할 수 있는 모델 아키텍처가 필요한 개발자 및 연구원.

주요 특징

• 다양한 모달리티: 텍스트 (ALBERT, RoBERTa, Llama 2, RWKV v7), 이미지 (ResNet, MobileNetV2, YOLOX, Stable Diffusion), 오디오/음성 (Whisper, Wav2Vec 2.0, OmniVoice) 모델을 포함합니다.
• 공식 및 커뮤니티 지원: 일련의 공식 Burn 모델과 커뮤니티에서 개발한 엄선된 구현을 결합합니다.
• 광범위한 응용 범위: 객체 탐지 및 OCR부터 음성 복제 및 악보 탐지까지 모든 것을 다룹니다.