File size: 1,151 Bytes
ee82736
 
 
 
 
 
 
7b14e5d
ee82736
 
 
83114ca
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1ffc364
 
 
 
 
 
 
 
 
 
83114ca
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
---
license: apache-2.0
language:
- en
metrics:
- accuracy
base_model:
- Inventic-AI/Account_Number_Extracter
pipeline_tag: image-to-text
tags:
- finance
---

## Description

This repository provides three YOLO-based models intended to be used sequentially to extract and recognize digits from document images.

The pipeline works in three stages:

1. **Region Segmentation**

   * `segmenter.pt` (YOLOv11n)
   * Finds the account number region in a document image
   * Output is cropped and passed to the next stage

2. **Digit Detection**

   * `BBox.pt` (YOLOv11n)
   * Detects bounding boxes for each digit within the cropped region
   * Bounding boxes should be sorted left-to-right

3. **Digit Classification**

   * `Classify.pt` (YOLOv11s-cls)
   * Classifies each cropped digit image into labels `0–9`
   * Predictions are concatenated to form the final sequence

### Sample Output:
Stage 1:

![image](https://cdn-uploads.huggingface.co/production/uploads/6970724f48a639796bed815e/zbG8OW4S3cYcDPR6MoD8-.png)

Stage 2:


![image](https://cdn-uploads.huggingface.co/production/uploads/6970724f48a639796bed815e/EufGV63YHKlabOlXEQJ9K.png)

---