File size: 778 Bytes
2e9091e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8aef362
0a071ab
 
2e9091e
 
 
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
---
language:
- fa
- en
library_name: hezar
pipeline_tag: object-detection
tags:
- hezar
---
## CRAFT: Character-Region Awareness For Text detection

CRAFT is a multilingual text detection model. The original implementation is located at https://github.com/clovaai/CRAFT-pytorch. This repo is only compatible 
with the [Hezar](https://github.com/hezarai/hezar) package.


### Usage
```
pip install hezar[vision]
```
```python
import cv2
from hezar.models import Model
from hezar.utils import draw_boxes


model = Model.load("hezarai/CRAFT", device="cuda")
outputs = model.predict("../assets/text_detection_example.png")
image = cv2.imread("../assets/text_detection_example.png")
result_image = draw_boxes(image, outputs[0]["boxes"])
cv2.imwrite("detected.png", result_image)
```