InterDigitalInc/CompressAI
A PyTorch library and evaluation platform for end-to-end compression research
What it solves
딥러닝을 활용한 엔드‑투‑엔드 데이터 압축을 연구·개발하기 위한 표준화된 PyTorch 기반 프레임워크를 제공합니다. 이를 통해 연구자는 기존 코덱을 넘어 이미지와 비디오에 대한 학습 기반 압축 모델을 만들 수 있습니다.
How it works
CompressAI는 학습 기반 압축을 위해 설계된 커스텀 PyTorch 연산, 레이어, 모델 모음을 제공합니다. TensorFlow compression 라이브러리의 일부를 포팅했으며, 바로 인코딩·디코딩에 사용할 수 있는 사전 학습 모델을 포함합니다. 또한 레이트‑왜곡 손실을 사용하는 학습 스크립트와 BPG, VTM, x265와 같은 전통 코덱과 학습 모델의 성능을 비교할 수 있는 평가 도구도 제공됩니다.
Who it’s for
데이터 압축, 컴퓨터 비전, 딥러닝을 전문으로 하는 AI 연구자 및 엔지니어로, 학습 기반 이미지·비디오 압축 모델을 훈련·평가·벤치마크할 플랫폼이 필요한 분들을 위한 것입니다.
Highlights
- End-to-End Research Platform: 딥러닝 기반 압축을 위한 훈련부터 평가까지의 완전 파이프라인.
- Pre-trained Model Zoo: 학습 이미지 압축을 위한 즉시 사용 가능한 사전 학습 모델에 접근.
- Benchmarking Tools: 학습 모델을 기존 이미지·비디오 압축 표준과 비교할 수 있는 내장 스크립트.
- Flexible Training: 커스텀 모델 구현 및 다중 GPU에 걸친 분산 훈련 지원.