ml-illustrated / efficientnet_lite2_pytorch_model

Pretrained Pytorch model file for EfficientNet Lite2

efficientnet-lite2-pytorch-model

Pretrained Pytorch model file for EfficientNet Lite2

Installation

pip install efficientnet_lite2_pytorch_model

Basic Usage

from efficientnet_lite2_pytorch_model import EfficientnetLite2ModelFile
print( 'model file path is %s' % ( EfficientnetLite2ModelFile.get_model_file_path() ) )

Actual Usage

Install the model package from EfficientNet-Lite-PyTorch:

pip install efficientnet_lite_pytorch

Load the model with pretrained weights:

from efficientnet_lite_pytorch import EfficientNet

from efficientnet_lite2_pytorch_model import EfficientnetLite2ModelFile
weights_path = EfficientnetLite2ModelFile.get_model_file_path()

lite2_model = EfficientNet.from_pretrained('efficientnet-lite2', weights_path = weights_path )

Credits

This package was created with Cookiecutter and the audreyr/cookiecutter-pypackage project template.

• Cookiecutter: https://github.com/audreyr/cookiecutter
• audreyr/cookiecutter-pypackage: https://github.com/audreyr/cookiecutter-pypackage
• cookiecutter-ml-model-files: https://github.com/ml-illustrated/cookiecutter-ml-model-files