首页 关于我们 成功案例 网络营销 电商设计 新闻中心 联系方式
QQ联系
电话联系
手机联系

【校园AI Day-AI workshop】自定义区域识别重命名

发布时间:2025-07-30 11:00
发布者:网络
浏览次数:
该项目针对人工摘录图片指定区域信息效率低的问题,利用PaddleHUB的OCR技术,对30张交付验收单图片进行处理。通过识别文本、计算坐标最小欧式距离提取铁塔名称,实现文件重命名,正确率达93.3%,可提升实际工作效率。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

【校园ai day-ai workshop】自定义区域识别重命名 -

基于OCR的自定义区域识别文件重命名

一、项目背景

  • 在实际的工作中,有时候需要根据图片中的指定区域进行重命名,人工摘录工作效率低。
  • 利用飞桨的PaddleHUB可实现自主框选识别区,实现识别内容的精确提取,从而提高效率。

二、项目目标

  • 通过PaddleHUB技术识别图片中的铁塔名称,并根据该字段完成对图片的重命名。

三、数据集

  • 本任务提供30张交付验收单,来自 https://aistudio.baidu.com/aistudio/datasetdetail/142101

四、项目实现思路

  1. 整体思路:利用hub.Module.recognize_text获得OCR结果,在此基础上,通过计算最小距离,获得表单相关信息(铁塔实际名称),完成文件重命名。
  2. 对每张图片(每个表单)进行OCR识别,并将结果保存至ocr_result文件夹。
  3. 根据先验知识,获取所需内容(铁塔实际名称)的中心位置,记为目标坐标。
  4. 对某个图片(表单),遍历每条识别结果,计算其中心坐标,计算该坐标和目标坐标之间的欧式距离
  5. 获得最小欧式距离及其对应的文字识别结果
  6. 根据上述结果,重命名图片文件

五、项目代码实现

In [ ]
# 安装依赖包 !pip uninstall paddlehub -y
!pip install paddlehub shapely pyclipper
%env CUDA_VISIBLE_DEVICES=0%matplotlib inline
    In [ ]
#  解压数据集! unzip -q /home/aistudio/data/data142101/Scan_0012_0004.zip -d /home/aistudio/images
    In [ ]
# 导入需要的包import paddlehub as hub ,cv2,os,csv,globimport os
    In [ ]
# 项目实现for i in range(1,31):    # 遍历数据集图片
    # image_path:图片路径
    if i < 10:
        image_path = "images/Scan_0012_000" + str(i) + ".jpg"
    else:
        image_path = "images/Scan_0012_00" + str(i) + ".jpg"
    # print(image_path)

    # 利用hub.Module.recognize_text获得OCR结果,并保存图片至ocr_result文件夹,其中,
    # [cv2.imread(image_path) for image_path in glob.glob(image_path)] 用于读取图片并存入列表
    # use_gpu=True 使用GPU加速
    # output_dir 为输出结果路径
    form = hub.Module(name="chinese_ocr_db_crnn_server").recognize_text(\
    [cv2.imread(image_path) for image_path in glob.glob(image_path)],use_gpu=True, \
    output_dir='ocr_result', visualization=True, box_thresh=0.5, text_thresh=0.5)    # 初始化最小距离及对应的文本
    min_dis = 9999
    min_text = 'none'
    
    # 对OCR后的结果,即某份表单,遍历每条数据
    # 每条数据(form_data)包括识别后的文字text,置信度confidence,坐标等,项目涉及text和坐标
    for form_data in form[0]['data']:    
        # 初始化坐标中心
        data_x = 0
        data_y = 0

        # 注意数据格式:纵坐标在前,即每对坐标第一个值为y,第二个值为x
        for j in range(4):
            data_y += form_data["text_box_position"][j][0]
            data_x += form_data["text_box_position"][j][1]        # 计算四个坐标平均值,即为对应文本框坐标中心
        data_x = data_x / 4
        data_y = data_y / 4

        # 坐标中心离目标中心最近的数据为所需数据,暂定目标中心[220,1290]
        # 计算每条数据坐标中心距离目标的欧式距离
        distance = ((data_x - 220) ** 2 + (data_y - 1290) ** 2) ** 0.5

        # 如果距离小于最小距离,更新最小距离及对应文本
        if distance <= min_dis:
            min_dis = distance
            min_text = form_data["text"]    print(i,image_path,min_text)    # 打印结果
    
    # 重命名文件
    os.rename(image_path, "images/" + min_text + '.jpg')
   

六、项目结果展示

  • 解压缩数据集后,重命名前: 【校园AI Day-AI workshop】自定义区域识别重命名 -            
  • OCR识别结果 【校园AI Day-AI workshop】自定义区域识别重命名 -            
  • 重命名后 【校园AI Day-AI workshop】自定义区域识别重命名 -            

七、项目总结

项目通过PaddleHUB技术识别图片中的铁塔名称,并根据该字段完成对图片的重命名,正确率达到93.3%(28/30)。 项目能够将PaddleHUB技术落地,在实际的工作中根据图片指定区域,对文件进行重命名,提高工作效率。

Motiff妙多 Motiff妙多

Motiff妙多是一款AI驱动的界面设计工具,定位为“AI时代设计工具”

Motiff妙多 334 查看详情 Motiff妙多

以上就是【校园AI Day-AI workshop】自定义区域识别重命名的详细内容,更多请关注其它相关文章!


# ai  # udio  # 重命名  # 自定义  # 中文网  # 表单  # 工作效率  # 每条  # 遍历  # 所需  # 系列产品  # 首款  # 枣庄建设电梯招标网站  # 论坛推广营销总结  # 通海集团网站建设  # 工商网站建设产品介绍  # 平谷区网站建设推广  # 成都seo优化套餐  # 建设行业网站推荐信  # 十堰百度seo优化  # 读书怎么找关键词排名  # 营销推广主线