ONNX 模型库
返回模型

说明文档

来自 Torchvision 的 MobileNet V2 模型,针对 Imagenette 数据集进行了微调。检查点使用 https://github.com/alexsu52/mobilenet_v2_imagenette 训练了 100 个 epoch。

Top-1 准确率为 98.64%。

主要目的是在模型优化的示例和演示中使用它。以下是优势:

  • Imagenette 可以自动下载,而且非常小。
  • MobileNet v2 易于训练且轻量级,同时也具有代表性,并被用于许多公开基准测试。

以下是在 PyTorch 中加载检查点的代码:

import torch
import torchvision.models as models

CHECKPOINT_URL = 'https://huggingface.co/alexsu52/mobilenet_v2_imagenette/resolve/main/pytorch_model.bin'
IMAGENETTE_CLASSES = 10

model = models.mobilenet_v2(num_classes=IMAGENETTE_CLASSES)  
        
checkpoint = torch.hub.load_state_dict_from_url(CHECKPOINT_URL, progress=False)
model.load_state_dict(checkpoint['state_dict'])

alexsu52/mobilenet_v2_imagenette

作者 alexsu52

↓ 0 ♥ 0

创建时间: 2022-12-05 05:50:21+00:00

更新时间: 2022-12-15 08:56:30+00:00

在 Hugging Face 上查看

文件 (6)

.gitattributes
README.md
mobilenet_v2_imagenette.onnx ONNX
openvino_model.tgz
pytorch_model.bin
tf_model.h5