kornia/kornia-rs

해결하는 문제

Kornia-rs는 Rust의 컴퓨터 비전 작업을 위한 고성능 저수준 기반을 제공합니다. 불필요한 데이터 복사 없이 원시 카메라 데이터와 PyTorch 또는 TensorRT와 같은 머신러닝 추론 엔진 사이의 간극을 원활하게 연결하는 빠르고 스레드 안전한 이미지 처리에 대한 요구를 해결합니다.

작동 방식

데이터가 위치한 곳에 따라 CPU 또는 NVIDIA GPU로 작업을 자동으로 분배하는 통합 API를 사용합니다. CPU 작업에는 SIMD (NEON/AVX2)를, GPU 작업에는 사용자 정의 CUDA 커널을 활용하여 실시간 파이프라인을 구현합니다. DLPack 및 CUDA Array Interface와 같은 표준을 통해 제로 카피 상호 운용성을 지원하며, 처리된 프레임을 딥러닝 모델로 직접 전달할 수 있습니다.

대상 사용자

이 라이브러리는 카메라-추론 파이프라인, 로봇 공학, 과학적 이미징 등 Rust의 메모리 안전성과 하드웨어 가속 처리의 원시 성능이 모두 필요한 실시간 컴퓨터 비전 애플리케이션을 구축하는 개발자를 위해 설계되었습니다.

주요 특징

• 메모리 및 스레드 안전성을 위해 Rust로 작성되었으며, 전역 인터프리터 락 (GIL)이 없습니다.
• CPU와 GPU 실행을 자동으로 처리하는 통합 API.
• PyTorch 및 TensorRT와의 제로 카피 통합.
• V4L2 카메라 캡처 및 전처리를 위한 융합 CUDA 커널을 통해 실시간 처리 지원.
• JPEG (libjpeg-turbo 사용), PNG, TIFF, WebP를 포함한 다양한 이미지 형식 지원.
• Python 바인딩을 제공하며, 표준 및 프리 스레드 (free-threaded) Python 빌드를 모두 지원합니다.