EfficientNet Lite2的预训练Pythorch模型文件
efficientnet-lite2-pytorch-model的Python项目详细描述
efficientnet-lite2-Pythorch模型
EfficientNet Lite2的预训练Pythorch模型文件
安装
pip install efficientnet_lite2_pytorch_model
基本用途
^{pr2}$实际使用量
从EfficientNet-Lite-PyTorch安装模型包:
pip install efficientnet_lite_pytorch
使用预训练权重加载模型:
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 )
学分
这个包是用Cookiecutter和audreyr/cookiecutter-pypackage
项目模板创建的。在
- Cookiecutter:https://github.com/audreyr/cookiecutter
audreyr/cookiecutter-pypackage
:https://github.com/audreyr/cookiecutter-pypackagecookiecutter-ml-model-files
:https://github.com/ml-illustrated/cookiecutter-ml-model-files
- 项目
标签: