File size: 1,114 Bytes
e4e565c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
---
license: mit
base_model:
- google/efficientnet-b0
---
# EfficientNet-B0 Document Image Classifier
This is an image classification model based on **Google EfficientNet-B0**, fine-tuned to classify input images into one of the following 16 categories:
1. **bar_chart**
2. **bar_code**
3. **chemistry_markush_structure**
4. **chemistry_molecular_structure**
5. **flow_chart**
6. **icon**
7. **line_chart**
8. **logo**
9. **map**
10. **other**
11. **pie_chart**
12. **qr_code**
13. **remote_sensing**
14. **screenshot**
15. **signature**
16. **stamp**
## Citation
If you use this model in your work, please cite the following papers:
```
@article{Tan2019EfficientNetRM,
title={EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks},
author={Mingxing Tan and Quoc V. Le},
journal={ArXiv},
year={2019},
volume={abs/1905.11946}
}
@techreport{Docling,
author = {Deep Search Team},
month = {8},
title = {{Docling Technical Report}},
url={https://arxiv.org/abs/2408.09869},
eprint={2408.09869},
doi = "10.48550/arXiv.2408.09869",
version = {1.0.0},
year = {2024}
}
``` |