dbnet轻型网络文本检测 - python 实现

DataBall 助力快速掌握数据集的信息和使用方式,会员享有 百种数据集,持续增加中。

需要更多数据资源和技术解决方案,知识星球: "DataBall - X 数据球(free)"


通过dbnet对图片进行文本框检测。

dbnet特点:

1) 一个轻型网络,cpu可以实时推理;

2)对图片的旋转文字进行定位;

3)对于图片的定位召回适中。

对于简单场景可以满足需求。

实现代码如下:

python 复制代码
import cv2
from models import DBNET


if __name__ == "__main__":
    text_handle = DBNET(MODEL_PATH="./models/dbnet.onnx")
    img = cv2.imread("images/2.jpg")
    # 文本检测
    box_list, score_list = text_handle.process(img,short_size=320)
    # 打印 置信度 和 对应的区域框
    print()
    for i in range(len(score_list)):
        print("[{}] 置信度 score:{:.2f} , 边界框 x1y1x2y2x3y3x4y4 :{}".format(i+1,score_list[i],box_list[i].reshape(-1)))
    # 显示文本检测框可视化结果
    img = draw_bbox(img, box_list)
    # 保存检测结果图
    cv2.imwrite("test_result.jpg", img)

    cv2.namedWindow("img",0)
    cv2.imshow("img",img)

    cv2.waitKey(0)

log显示如下:

python 复制代码
[1] 置信度 score:0.74 , 边界框 x1y1x2y2x3y3x4y4 :[118 290 213 247 232 288 136 332]
[2] 置信度 score:0.76 , 边界框 x1y1x2y2x3y3x4y4 :[109 260 203 223 216 256 122 293]
[3] 置信度 score:0.72 , 边界框 x1y1x2y2x3y3x4y4 :[100 236 187 194 201 224 114 266]
[4] 置信度 score:0.74 , 边界框 x1y1x2y2x3y3x4y4 :[ 87 205 176 169 188 198  98 233]
[5] 置信度 score:0.71 , 边界框 x1y1x2y2x3y3x4y4 :[ 69 168 168 133 182 174  84 208]

示例如下:

​​​​

助力快速掌握数据集的信息和使用方式。

数据可以如此美好!

相关推荐
我爱一条柴ya8 分钟前
【AI大模型】神经网络反向传播:核心原理与完整实现
人工智能·深度学习·神经网络·ai·ai编程
万米商云12 分钟前
企业物资集采平台解决方案:跨地域、多仓库、百部门——大型企业如何用一套系统管好百万级物资?
大数据·运维·人工智能
新加坡内哥谈技术16 分钟前
Google AI 刚刚开源 MCP 数据库工具箱,让 AI 代理安全高效地查询数据库
人工智能
慕婉030717 分钟前
深度学习概述
人工智能·深度学习
大模型真好玩18 分钟前
准确率飙升!GraphRAG如何利用知识图谱提升RAG答案质量(额外篇)——大规模文本数据下GraphRAG实战
人工智能·python·mcp
198919 分钟前
【零基础学AI】第30讲:生成对抗网络(GAN)实战 - 手写数字生成
人工智能·python·深度学习·神经网络·机器学习·生成对抗网络·近邻算法
6confim19 分钟前
AI原生软件工程师
人工智能·ai编程·cursor
阿里云大数据AI技术20 分钟前
Flink Forward Asia 2025 主旨演讲精彩回顾
大数据·人工智能·flink
i小溪21 分钟前
在使用 Docker 时,如果容器挂载的数据目录(如 `/var/moments`)位于数据盘,只要服务没有读写,数据盘是否就不会被唤醒?
人工智能·docker
程序员NEO23 分钟前
Spring AI 对话记忆大揭秘:服务器重启,聊天记录不再丢失!
人工智能·后端