PINTO0309/PINTO_model_zoo

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

What it solves

PINTO_model_zoo는 사전 변환 및 사전 양자화된 AI 모델을 중앙 집중식으로 모아 제공합니다. 서로 다른 딥러닝 프레임워크 간에 모델을 수동으로 변환하고 엣지 디바이스(예: Raspberry Pi 또는 EdgeTPU)용으로 최적화하는 어려움을 해결하며, 여러 포맷으로 바로 사용할 수 있는 인기 모델 버전을 제공합니다.

How it works

이 프로젝트는 다양한 프레임워크 간에 상호 변환된 모델들의 "동물원"을 유지합니다. 지원 포맷은 TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlow Lite(Float32, Float16, INT8), EdgeTPU, CoreML 등을 포함합니다. 저장소에는 이미지 분류, 2D 객체 탐지 및 기타 비전 작업용 모델이 포함되어 있으며, Weight, Integer, Full Integer, Dynamic Range와 같은 다양한 양자화 레벨을 제공해 성능과 정확도 사이의 균형을 맞춥니다.

Who it’s for

엣지 AI 및 모바일 배포를 진행하는 개발자와 연구자를 위한 것으로, 특정 포맷의 최적화된 모델이 필요하지만 복잡한 변환 및 양자화 과정을 직접 수행하고 싶지 않은 경우에 적합합니다.

Highlights

  • Broad Framework Support: 거의 모든 주요 AI 프레임워크와 배포 포맷(TFLite, ONNX, CoreML, OpenVINO 등)을 지원합니다.
  • Extensive Model Library: 이미지 분류와 객체 탐지를 위한 방대한 사전 양자화 모델을 포함합니다.
  • Edge-Optimized: Raspberry Pi 4/3 및 EdgeTPU와 같은 하드웨어에서의 성능을 목표로 최적화되었습니다.
  • Quantization Variety: FP32, FP16, INT8, Dynamic Range 양자화 모델을 제공하여 다양한 하드웨어 제약에 맞출 수 있습니다.