|
|
--- |
|
|
license: apache-2.0 |
|
|
pipeline_tag: image-to-text |
|
|
tags: |
|
|
- layout |
|
|
--- |
|
|
|
|
|
## anydoclayout |
|
|
> docs layout detection |
|
|
|
|
|
<a href="https://huggingface.co/anyforge/anydoclayout" target="_blank"><img src="https://img.shields.io/badge/%F0%9F%A4%97-HuggingFace-blue"></a> |
|
|
<a href="https://www.modelscope.cn/models/anyforge/anydoclayout" target="_blank"><img alt="Static Badge" src="https://img.shields.io/badge/%E9%AD%94%E6%90%AD-ModelScope-blue"></a> |
|
|
<a href=""><img src="https://img.shields.io/badge/Python->=3.6-aff.svg"></a> |
|
|
<a href=""><img src="https://img.shields.io/badge/OS-Linux%2C%20Win%2C%20Mac-pink.svg"></a> |
|
|
|
|
|
``` |
|
|
_ ____ _ _ |
|
|
/ \ _ __ _ _| _ \ ___ ___| | __ _ _ _ ___ _ _| |_ |
|
|
/ _ \ | '_ \| | | | | | |/ _ \ / __| | / _` | | | |/ _ \| | | | __| |
|
|
/ ___ \| | | | |_| | |_| | (_) | (__| |__| (_| | |_| | (_) | |_| | |_ |
|
|
/_/ \_\_| |_|\__, |____/ \___/ \___|_____\__,_|\__, |\___/ \__,_|\__| |
|
|
|___/ |___/ |
|
|
|
|
|
``` |
|
|
|
|
|
- Github：[anydoclayout](https://github.com/anyforge/anydoclayout) |
|
|
- Hugging Face: [anydoclayout](https://huggingface.co/anyforge/anydoclayout) |
|
|
- ModelScope: [anydoclayout](https://www.modelscope.cn/models/anyforge/anydoclayout) |
|
|
|
|
|
 |
|
|
|
|
|
## train datasets |
|
|
|
|
|
### 1. labels info |
|
|
```python |
|
|
{0: 'header', |
|
|
1: 'title', |
|
|
2: 'text', |
|
|
3: 'table', |
|
|
4: 'figure', |
|
|
5: 'formula', |
|
|
6: 'footer', |
|
|
7: 'pagenum'} |
|
|
``` |
|
|
|
|
|
### 2. datasets info |
|
|
|
|
|
- train: 841862 (opendata: 667426, selfgen: 174436) |
|
|
- eval: 5500 |
|
|
- imgsize：1280 |
|
|
|
|
|
### 3. eval results |
|
|
```python |
|
|
Class Images Instances Box(P R) |
|
|
all 5500 52274 0.921 0.897 |
|
|
header 1461 2337 0.92 0.878 |
|
|
title 2308 5473 0.896 0.893 |
|
|
text 4149 34156 0.937 0.927 |
|
|
table 1476 1913 0.946 0.958 |
|
|
figure 1842 3343 0.94 0.94 |
|
|
formula 735 1506 0.881 0.876 |
|
|
footer 745 1157 0.909 0.781 |
|
|
pagenum 2164 2389 0.938 0.919 |
|
|
|
|
|
``` |
|
|
|
|
|
### if you want to get datasets |
|
|
- email：christnowx@qq.com |
|
|
|
|
|
|
|
|
### how to use |
|
|
|
|
|
```python |
|
|
from pathlib import Path |
|
|
from ultralytics import YOLO |
|
|
|
|
|
modelfile = Path(model_dir).joinpath('anydoclayout-yolo11s-imgsz1280.pt') |
|
|
model = YOLO(modelfile) |
|
|
res = model.predict('your img file', imgsz = 1280) |
|
|
|
|
|
``` |
|
|
|
|
|
### Buy me a coffee |
|
|
|
|
|
- 微信(WeChat) |
|
|
|
|
|
<div align="left"> |
|
|
<img src="./zanshan.jpg" width="30%" height="30%"> |
|
|
</div> |
|
|
|
