|
--- |
|
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} |
|
} |
|
``` |
