hezarai
/

CRAFT

arxyzan commited on Jul 4

Commit

0a071ab

•

1 Parent(s): ff7342b

Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -24,8 +24,8 @@ from hezar.utils import draw_boxes
 model = Model.load("hezarai/CRAFT", device="cuda")
-outputs = model.predict("../assets/sample.png")
-image = cv2.imread("../assets/O1.png")
 result_image = draw_boxes(image, outputs[0]["boxes"])
 cv2.imwrite("detected.png", result_image)
 ```

 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)
 ```