Image Classification
ONNX
efficientnet-b0 / README.md
qmller's picture
Update README.md
02c6e46 verified
---
license: apache-2.0
datasets:
- ILSVRC/imagenet-1k
pipeline_tag: image-classification
---
# Introduction
This repository stores the model for Efficientnet-b0, compatible with Kalray's neural network API. </br>
Please see www.github.com/kalray/kann-models-zoo for details and proper usage. </br>
Please see https://huggingface.co/docs/transformers/main/en/model_doc/efficientnet for Efficientnet model description. </br>
# Contents
- ONNX: efficientNet-b0.onnx
- Quantized ONNX (INT8): efficientNet-b0-q.onnx
# Lecture note reference
- EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks, https://arxiv.org/pdf/1905.11946
# Repository or links references
- [PyTorch | TorchVision](https://pytorch.org/vision/stable/models/generated/torchvision.models.efficientnet_b0.html#torchvision.models.efficientnet_b0)
BibTeX entry and citation info
```
@article{DBLP:journals/corr/abs-1905-11946,
author = {Mingxing Tan and Quoc V. Le},
title = {EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks},
journal = {CoRR},
volume = {abs/1905.11946},
year = {2019},
url = {http://arxiv.org/abs/1905.11946},
eprinttype = {arXiv},
eprint = {1905.11946},
timestamp = {Mon, 03 Jun 2019 13:42:33 +0200},
biburl = {https://dblp.org/rec/journals/corr/abs-1905-11946.bib},
bibsource = {dblp computer science bibliography, https://dblp.org}
}
```
Authors:
+ qmuller@kalrayinc.com
+ nbouberbachene@kalrayinc.com