metadata
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 package.
Usage
pip install hezar[vision]
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)