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.

mammoclip

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

efficientnet-lite-pytorch

EfficientNet-Lite implemented in PyTorch.

tvmodels

Implementation of vision models with their pretrained weights

timm-3d

PyTorch Volume Models for 3D data

efficientunet-pytorch

A PyTorch 1.0 Implementation of Unet with EfficientNet as encoder

segmentation-models-zwy

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

timm-ctp

(Unofficial) PyTorch Image Models

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

tensornets

High level network definitions with pre-trained weights in TensorFlow

hyperion-ml

Python toolkit for speech processing

efficientunet

Keras Implementation of Unet with EfficientNet as encoder

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

instances

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