pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

created at Feb. 2, 2019, 5:51 a.m.

Python

293

23,189

3,837

GitHub
gen-efficientnet-pytorch

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

created at May 11, 2019, 7:35 p.m.

Python

35

610

71

GitHub
efficientdet-pytorch

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

created at March 26, 2020, 11:09 p.m.

Python

22

372

35

GitHub