Model Zoo Python Packages

nnef

The NNEF Tools repository contains tools to generate and consume NNEF documents

nnef-tools

The NNEF Tools repository contains tools to generate and consume NNEF documents

robustbench

RobustBench: a standardized adversarial robustness benchmark [NeurIPS 2021 Benchmarks and Datasets Track]

depth-estimation

Unified Python library for monocular depth estimation with one API for inference, video streaming, visualization, evaluation, fine-tuning, and dataset loading.

pyrafuse

DINOv3-based semantic segmentation for skin, fabric, and background — PyTorch, Hugging Face, and TensorRT.

kimm

A library that includes Keras3 layers, blocks and models with pretrained weights, providing support for transfer learning, feature extraction, and more.

deepsafe-bench

Tests how well deepfake detectors actually work. 24 detectors scored against 411 AI generators: they catch 66% of fakes overall, but only 7% of Sora video. Successor to siddharthksah/DeepSafe.

mozo

Run several computer vision models or a graph of them from one pip install, with the original's exact numbers.

caffe2onnx

Convert Caffe models to ONNX.

openmodelzoo-modelapi

Pre-trained Deep Learning models and demos (high quality and extremely fast)

forest-utils

Pypi Package for Smoketrees model zoo

coreai-cli

Downloadable models and conversion recipes for Apple's Core AI on iPhone and Mac. Chat, vision, speech and generative models with per-model validation records, Swift examples through CoreAIKit, and a downloadable Mac app.

modelfeast

Pytorch model zoo for human, include all kinds of 2D CNN, 3D CNN, and CRNN