[P] EfficientNet in PyTorch
TL;DR: Implemented EfficientNet in PyTorch: https://github.com/lukemelas/EfficientNet-PyTorch
I found EfficientNet to be quite an exciting paper, so I implemented it in PyTorch over the past two days. The implementation makes it simple to load pretrained models and integrate them into your own projects/research/models. GitHub: https://github.com/lukemelas/EfficientNet-PyTorch
There are multiple examples in the GitHub repo and here is one on Colab. It’s as quick as
from efficientnet_pytorch import EfficientNet model = EfficientNet.from_pretrained('efficientnet-b0')
And you can install it via pip if you would like:
pip install efficientnet_pytorch
Finally, there are scripts to evaluate on ImageNet (with training scripts coming soon) and there’s functionality to easily extract image features.
Let me know (either here or on GitHub) if you have any comments or find any bugs. I hope some of you find it useful!
submitted by /u/LukeAndGeorge