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

关于目标分类图像增强界面、功能及代码框架

追梦IT人 2025-05-07
142

目标分类图像增强

功能介绍:目标分类图像增强功能对目标分类的图像数据进行增强处理,通过增加数据的多样性,提高模型的泛化能力,使模型能够更好地应对不同场景和条件下的图像分类任务。


操作步骤

  1. 进入图像增强界面:在平台主界面中,找到并点击目标分类图像增强功能入口,进入图像增强界面。界面主要分为文件夹设置区和图形增强功能区,操作与目标检测图像增强类似。

  2. 设置文件夹路径:在文件夹设置区,点击“设定文件夹”按钮,分别为“数据集图片文件夹”选择路径。选择后,相应文件夹的全路径将显示在对应的主界面上。

  3. 执行图像增强:点击“数据增强”按钮后,系统会根据您选择的图像增强操作,在数据集子图片文件夹下新增对应的增强图像文件和增强后的标注文件。在执行过程中,会弹出一个进度条窗口,显示转换的进度。完成后,系统会提示已完成图像增强,并自动返回主界面。

完成图像增强后的dog和horse子目录下如下:

目标分类图像增强界面代码层级关系

- ImageEnhancementThread
  - __init__: 初始化线程,接收图像文件夹和复选框信息
  - run: 线程运行时执行的主逻辑
  - read_labels: 读取标签文件
  - save_enhanced: 保存增强后的图像
  - random_rotation: 对图像进行随机旋转
  - horizontal_flip: 对图像进行水平翻转
  - vertical_flip: 对图像进行垂直翻转
  - random_crop: 对图像进行随机裁剪
  - center_crop: 对图像进行中心裁剪
  - resize: 对图像进行缩放
  - gray_transform: 对图像进行灰度转换
  - brightness_adjust: 对图像进行亮度调整
  - contrast_adjust: 对图像进行对比度调整
  - normalization: 对图像进行归一化处理
  - gaussian_noise: 为图像添加高斯噪声
  - gaussian_blur: 对图像进行高斯模糊
  - random_block: 对图像进行随机块处理
  - center_region: 提取图像的中心区域
- ImageClassficationEnhancementUI
  - __init__: 初始化用户界面类
  - initUI: 初始化用户界面
  - select_image_folder: 选择图像文件夹
  - select_label_folder: 选择标签文件夹
  - start_enhancement: 开始图像增强操作
  - on_enhancement_finished: 图像增强完成后的处理


目标分类图像增强界面功能总结

  1. ImageEnhancementThread
    :继承自QThread
    ,用于在后台线程中执行图像增强操作。

    • __init__
      :初始化线程,接收图像文件夹和复选框信息。
    • run
      :线程运行时执行的主逻辑,负责调用各种图像增强方法。
    • read_labels
      :读取标签文件。
    • save_enhanced
      :保存增强后的图像。
    • random_rotation
      :对图像进行随机旋转。
    • horizontal_flip
      :对图像进行水平翻转。
    • vertical_flip
      :对图像进行垂直翻转。
    • random_crop
      :对图像进行随机裁剪。
    • center_crop
      :对图像进行中心裁剪。
    • resize
      :对图像进行缩放。
    • gray_transform
      :对图像进行灰度转换。
    • brightness_adjust
      :对图像进行亮度调整。
    • contrast_adjust
      :对图像进行对比度调整。
    • normalization
      :对图像进行归一化处理。
    • gaussian_noise
      :为图像添加高斯噪声。
    • gaussian_blur
      :对图像进行高斯模糊。
    • random_block
      :对图像进行随机块处理。
    • center_region
      :提取图像的中心区域。
  2. ImageClassficationEnhancementUI
    :继承自QWidget
    ,用于创建用户界面。

    • __init__
      :初始化用户界面类。
    • initUI
      :初始化用户界面,设置窗口标题等。
    • select_image_folder
      :选择图像文件夹。
    • select_label_folder
      :选择标签文件夹。
    • start_enhancement
      :开始图像增强操作,启动ImageEnhancementThread
      线程。
    • on_enhancement_finished
      :图像增强完成后的处理,如显示提示信息等。
  3. 主程序:创建QApplication
    实例,设置样式表,创建ImageClassficationEnhancementUI
    实例并显示,最后进入应用程序的事件循环。


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

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

评论