Search Packages

timm

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

facenet-pytorch

Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models

pretrainedmodels

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

mmcls

OpenMMLab Pre-training Toolbox and Benchmark

resnest

ResNeSt: Split-Attention Networks

image-classifiers

Classification models trained on ImageNet. Keras.

mmpretrain

OpenMMLab Pre-training Toolbox and Benchmark

segmentation-models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

flaxmodels

Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

wespeakerruntime

Research and Production Oriented Speaker Verification, Recognition and Diarization Toolkit

cellfinder-core

Standalone cellfinder cell detection algorithm

sparsezoo

Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes

mindcv

A toolbox of vision models and algorithms based on MindSpore

xtreme-vision

A High Level Python Library to empower students, developers to build applications and systems enabled with computer vision capabilities.

mambular

DeepTab is a Python package that simplifies tabular deep learning by providing a suite of models for regression, classification, and distributional regression tasks. It includes models such as Mambular, TabM, FT-Transformer, TabulaRNN, TabTransformer, and tabular ResNets.

numpy-ml

Machine learning, in numpy

cellfinder-napari

Efficient cell detection in large images using cellfinder in napari

jai

Just Assemble IT! - A LEGO-style & PyTorch-based Deep Learning Library

pytorch2keras

The deep learning models converter

image-classification-jax

Image classification in JAX with ViT, resnet, cifar10, cifar100, imagenette, and imagenet

timm-ctp

(Unofficial) PyTorch Image Models

tvmodels

Pytorch implementation of vision models.

deepvision-toolkit

PyTorch and TensorFlow/Keras image models with automatic weight conversions and equal API/implementations - Vision Transformer (ViT), ResNetV2, EfficientNetV2, NeRF, SegFormer, MixTransformer, (planned...) DeepLabV3+, ConvNeXtV2, YOLO, etc.

neural-dream

A PyTorch implementation of DeepDream