hezarai/CRAFT

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 (>=v0.40.0).

Usage

pip install hezar[vision]

from hezar.models import Model
from hezar.utils import load_image, draw_boxes, show_image


model = Model.load("hezarai/CRAFT", device="cuda")
image = load_image("../assets/text_detection_example.jpg")
outputs = model.predict(image)
result_image = draw_boxes(image, outputs[0]["boxes"])
show_image(result_image, "text_detected")