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

effdet

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

efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

geffnet

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

segmentation-models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

timm-3d

PyTorch Volume Models for 3D data

timm-ctp

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

tensornets

High level network definitions with pre-trained weights in TensorFlow

mammoclip

[MICCAI 2024, top 11%] Official Pytorch implementation of Mammo-CLIP: A Vision Language Foundation Model to Enhance Data Efficiency and Robustness in Mammography

hyperion-ml

Python toolkit for speech processing

efficientnet-lite-pytorch

Pytorch implementation of EfficientNet Lite variants

fastseg

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

timm-clean

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

tvmodels

Pytorch implementation of vision models.

efficientunet-pytorch

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

efficientunet

Keras Implementation of Unet with EfficientNet as encoder

segmentation-models-zwy

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

timm-eidl

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