| --- |
| license: apache-2.0 |
| datasets: |
| - imagenet-1k |
| metrics: |
| - accuracy |
| tags: |
| - RyzenAI |
| - vision |
| - classification |
| - pytorch |
| --- |
| |
| # Inception_v4 |
| Quantized Inception_v4 model that could be supported by [AMD Ryzen AI](https://ryzenai.docs.amd.com/en/latest/). |
|
|
|
|
| ## Model description |
| Inception_v4 was first introduced in the paper [Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning](https://arxiv.org/abs/1602.07261). |
| |
| The model implementaion is from [TensorFlow-Slim](https://github.com/tensorflow/models/tree/master/research/slim). |
| |
| |
| ## How to use |
| |
| ### Installation |
| |
| Follow [Ryzen AI Installation](https://ryzenai.docs.amd.com/en/latest/inst.html) to prepare the environment for Ryzen AI. |
| Run the following script to install pre-requisites for this model. |
| |
| ```bash |
| pip install -r requirements.txt |
| ``` |
| |
| ### Data Preparation |
| |
| Follow [imagenet-1k](https://huggingface.co/datasets/imagenet-1k) to download dataset. |
| |
| Download [ImageNet validation synset labels file](https://github.com/tensorflow/models/blob/master/research/slim/datasets/imagenet_2012_validation_synset_labels.txt). |
| |
| Create validation image list: |
| ```bash |
| python create_image_list.py imagenet_2012_validation_synset_labels.txt |
| ``` |
| |
| ### Model Evaluation |
| |
| ```python |
| 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 |
| ``` |
| |
| ### Performance |
| |
| |Metric |Accuracy on IPU| |
| | :----: | :----: | |
| |Top1/Top5| 79.92% / 95.02%| |
| |
| |
| ```bibtex |
| @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}, |
| } |
|
|
| ``` |