license: apache-2.0 | |
# Suryolo : Layout Model For Arabic Documents | |
Suryolo is combination of Surya layout Model form SuryaOCR(based on Segformer) and YoloV10 objection detection. | |
## Setup Instructions | |
### Clone the Surya OCR GitHub Repository | |
```bash | |
git clone https://github.com/vikp/surya.git | |
cd surya | |
``` | |
### Switch to v0.4.14 | |
```bash | |
git checkout f7c6c04 | |
``` | |
### Install Dependencies | |
You can install the required dependencies using the following command: | |
```bash | |
pip install -r requirements.txt | |
``` | |
```bash | |
pip install ultralytics | |
``` | |
```bash | |
pip install supervision | |
``` | |
### Suryolo Pipeline | |
Download `surya_yolo_pipeline.py` file from the Repository. | |
```python | |
from surya_yolo_pipeline import suryolo | |
from surya.postprocessing.heatmap import draw_bboxes_on_image | |
image_path = "sample.jpg" | |
image = Image.open(image_path) | |
bboxes = suryolo(image_path) | |
plotted_image = draw_bboxes_on_image(bboxes,image) | |
``` | |
#### Refer to `benchmark.ipynb` for comparison between Traditional Surya Layout Model and Suryolo Layout Model. |