提交 59d0a00e 编写于 作者: L LDOUBLEV

opt visualized func and add docker usage in cpu

上级 4af18109
...@@ -276,14 +276,6 @@ def text_visual(texts, scores, img_h=400, img_w=600, threshold=0.): ...@@ -276,14 +276,6 @@ def text_visual(texts, scores, img_h=400, img_w=600, threshold=0.):
if __name__ == '__main__': if __name__ == '__main__':
text = [
"旨在打造一套丰富领先、且实用的工具库助力使、用者训练出更好的模型,并应用落地",
"以下代码实现了文本检测、识别串联推理,在执行预测时,需要通过参数image_dir指定单张图像或者图像集合",
"上述DB模型的训练和评估,需设置后处理参数box_thresh=0.6,unclip_ratio=1.5,使用不同数据集"
]
img = text_visual(text, scores=[0.999, 0.999, 0.999], img_h=100)
cv2.imwrite("./draw_txt.jpg", np.array(img))
"""
test_img = "./doc/test_v2" test_img = "./doc/test_v2"
predict_txt = "./doc/predict.txt" predict_txt = "./doc/predict.txt"
f = open(predict_txt, 'r') f = open(predict_txt, 'r')
...@@ -303,4 +295,3 @@ if __name__ == '__main__': ...@@ -303,4 +295,3 @@ if __name__ == '__main__':
new_img = draw_ocr(image, boxes, txts, scores, draw_txt=True) new_img = draw_ocr(image, boxes, txts, scores, draw_txt=True)
cv2.imwrite(img_name, new_img) cv2.imwrite(img_name, new_img)
"""
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册