暂无图片
暂无图片
暂无图片
暂无图片
暂无图片

关于目标检测预测界面、功能及代码框架

追梦IT人 2025-04-28
224

目标检测预测

功能介绍:目标检测预测功能使用训练好的目标检测模型对图片或视频进行分析,识别其中的目标物体,并将检测结果以可视化的方式展示出来。用户可以通过该功能评估模型的性能,验证模型在实际应用中的效果。

操作步骤

  1. 进入预测界面:在平台主界面中,找到并点击目标检测预测功能入口,进入预测界面。界面主要分为图片操作区、检测结果区和检测结果与位置信息区。

  2. 设置数据模型:在图片操作区,点击选择数据模型,打开之前保存好的pt文件

  3. 选择待检测的图片或视频


    • 打开图片
      :点击“打开图片”按钮,在弹出的文件选择窗口中选择要检测的图片文件。选择后,图片会在左侧的图片预览区缩小展示,图片地址会显示在左侧的图片地址输入框中,中间的媒体浏览区会展示图片的原图。同时,系统会调用选择的模型进行预测,预测结果的结构化数据会输出到检测结果与位置信息区,媒体浏览区会显示所有预测标注框,并且默认选中第一行预测标注框,该标注框会在媒体浏览区高亮显示,同时在检测结果区的当前目标栏显示其类型和置信度,在目标位置栏分别显示预测框的边界值。
    • 打开文件夹
      :点击“打开文件夹”按钮,选择包含图片的文件夹。系统会读取该文件夹中的第一张图片,在左侧的图片预览区显示图片缩略图,在左侧的图片地址输入框中显示图片地址,在中间的媒体浏览区展示图片。同时,系统会调用模型进行预测,将预测结果输出到检测结果与位置信息区,并默认选中第一行预测标注框,在媒体浏览区显示所有预测标注框,在检测结果区的当前目标栏显示其类型和置信度,在目标位置栏分别显示预测框的边界值。
    • 打开视频
      :点击“打开视频”按钮,在弹出的文件选择窗口中选择要检测的视频文件。选择后,视频的图片帧会在中间的媒体浏览区进行展示,同时系统会调用模型进行预测,将预测结果直接输出到媒体浏览区,检测结果区和检测结果与位置信息区不进行展示。
    • 打开摄像头
      :点击“打开摄像头”按钮,电脑的摄像头会开启,摄像头捕获的图片帧会在中间的媒体浏览区展示,同时系统会调用模型进行预测,将预测结果输出到媒体浏览区,检测结果区和检测结果与位置信息区不进行展示。
  4. 调整检测参数:在检测结果区,有检测参数设置部分,包括置信度阈值和交并比阈值。


    • 置信度阈值
      :默认值为 0.5,可调节到小数位。该参数用于过滤掉置信度较低的预测结果,只有置信度高于该阈值的预测框才会被保留。
    • 交并比阈值
      :默认值为 0.4,可调节到小数位。该参数用于非极大值抑制(NMS)操作,用于去除重叠度较高的预测框,只保留得分最高的预测框。
  5. 查看检测结果:检测结果与位置信息区显示预测结果,点击其中一行,媒体浏览区会高亮显示对应的预测框,同时更新检测结果区的当前目标和目标位置信息。

  6. 检测结果区:检测结果区展示基于模型预测出来的信息总数和预测时间。

  7. 当前目标区:当前目标区展示检测结果与位置信息区选中的预测结果,包括预测目标类型、置信度、左上、右下的坐标。

目标检测预测界面代码层级关系与功能说明

模块导入

导入所需的库,包括操作系统相关库、计算机视觉库、PyQt5 库以及 YOLO 模型库等。

CustomObjectDetectionApp

这是一个继承自QWidget
的自定义类,用于创建一个自定义的目标检测应用程序界面。


  • __init__
    :类的构造函数,初始化对象,但函数体省略。
  • initUI
    :初始化用户界面,设置窗口的样式、布局等。
    • 注释部分设置了窗口各组件的样式表。
    • 主布局采用纵向布局,但具体实现省略。
  • set_model
    :打开文件选择对话框,用于选择模型文件,但当前实现省略了停止当前媒体操作的功能。
  • create_image_operation_area
    :创建图片操作区的布局,具体实现省略。
  • create_detection_result_area
    :创建检测结果区的布局,新增了模型介绍区域,具体实现省略。
  • create_detection_operation_area
    :创建检测操作区的布局,具体实现省略。
  • open_image
    :打开文件选择对话框,用于选择图片文件,具体实现省略。
  • open_folder
    :打开文件夹选择对话框,用于选择文件夹,具体实现省略。
  • open_video
    :打开文件选择对话框,用于选择视频文件,具体实现省略。
  • open_camera
    :打开摄像头,具体实现省略。
  • clear_data
    :清除数据,具体实现省略。
  • stop_current_media_operation
    :停止当前的媒体操作,具体实现省略。
  • save_results
    :保存检测结果,但当前功能暂未实现。
  • show_image_preview
    :显示图片预览,具体实现省略。
  • detect_image
    :开始对图片进行检测,并计时,具体实现省略。
  • update_video_frame
    :更新视频帧,具体实现省略。
  • show_image
    :显示图片,具体实现省略。
  • prev_image
    :显示上一张图片,具体实现省略。
  • next_image
    :显示下一张图片,具体实现省略。
  • zoom_in
    :放大图片,并增加放大限制,具体实现省略。
  • zoom_out
    :缩小图片,并增加缩小限制,具体实现省略。
  • update_image_display
    :更新图片显示,具体实现省略。
  • handle_wheel_event
    :处理鼠标滚轮事件,具体实现省略。
  • table_row_clicked
    :处理表格行点击事件,具体实现省略。


目标检测预测界面代码层级关系

- 模块导入
- 类 CustomObjectDetectionApp
  - __init__
  - initUI
  - set_model
  - create_image_operation_area
  - create_detection_result_area
  - create_detection_operation_area
  - open_image
  - open_folder
  - open_video
  - open_camera
  - clear_data
  - stop_current_media_operation
  - save_results
  - show_image_preview
  - detect_image
  - update_video_frame
  - show_image
  - prev_image
  - next_image
  - zoom_in
  - zoom_out
  - update_image_display
  - handle_wheel_event
  - table_row_clicked
- 主程序入口


最后欢迎关注公众号:python与大数据分析

文章转载自追梦IT人,如果涉嫌侵权,请发送邮件至:contact@modb.pro进行举报,并提供相关证据,一经查实,墨天轮将立刻删除相关内容。

评论