ONNX 模型库
返回模型

说明文档

Inception_v4

量化版 Inception_v4 模型,可由 AMD Ryzen AI 支持。

模型描述

Inception_v4 首次发表于论文 Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning。

该模型的实现来自 TensorFlow-Slim。

使用方法

安装

按照 Ryzen AI 安装指南 准备 Ryzen AI 环境。 运行以下脚本安装此模型的依赖项。

pip install -r requirements.txt

数据准备

按照 imagenet-1k 下载数据集。

下载 ImageNet 验证集 synset 标签文件。

创建验证图像列表:

python create_image_list.py imagenet_2012_validation_synset_labels.txt

模型评估

python eval_onnx.py --onnx_model inceptionv4_int8.onnx --ipu --provider_config Path\To\vaip_config.json --val_data_dir /Path/To/Your/Validation/Data --val_image_list val.txt

性能

指标 IPU 上的准确率
Top1/Top5 79.92% / 95.02%
@article{Szegedy2016Inceptionv4IA,
  title={Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning},
  author={Christian Szegedy and Sergey Ioffe and Vincent Vanhoucke and Alexander A. Alemi},
  journal={arXiv:1602.07261},
  year={2016},
}

amd/inception_v4

作者 amd

↓ 0 ♥ 0

创建时间: 2023-12-05 13:01:15+00:00

更新时间: 2023-12-06 09:37:23+00:00

在 Hugging Face 上查看

文件 (6)

.gitattributes
README.md
create_image_list.py
eval_onnx.py
inceptionv4_int8.onnx ONNX
requirements.txt