Efficientnet Python 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

13.3M 37K 5K

effdet

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights

562K 2K 302

efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

84K 2K 463

geffnet

Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS

55K 2K 216

segmentation-models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

19K 5K 1K

mammoclip

Mammo-CLIP: A Vision Language Foundation Model to Enhance Data Efficiency and Robustness in Mammography

1K 90 34

timm-3d

Python library with Neural Networks for Volume (3D) Classification based on PyTorch.

874 84 8

efficientnet-lite-pytorch

EfficientNet-Lite implemented in PyTorch.

666 15 10

tvmodels

Pytorch implementation of vision models.

644 12 1

timm-ctp

(Unofficial) PyTorch Image Models

543 37K 5K

fastseg

📸 PyTorch implementation of MobileNetV3 for real-time semantic segmentation, with pretrained weights & state-of-the-art performance

423 362 34

efficientunet-pytorch

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

402 161 34

hyperion-ml

Python toolkit for speech processing

296 72 21

tensornets

High level network definitions with pre-trained weights in TensorFlow

277 1K 179

timm-clean

(Unofficial) PyTorch Image Models, with torch dependency removed in setup.py

269 37K 5K

segmentation-models-zwy

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

226 5K 1K

efficientunet

Keras Implementation of Unet with EfficientNet as encoder.

208 48 26

timm-eidl

PyTorch Image Models

76 37K 5K

instances

1 37K 5K