返回模型
说明文档
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 下载数据集。
创建验证图像列表:
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