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

关于目标分类训练界面、功能及代码框架

追梦IT人 2025-05-08
71

目标分类训练

功能介绍:目标分类训练利用标注好的图像数据训练分类模型,通过调整训练参数,使模型能够准确地对新的图像进行分类。


操作步骤

  1. 进入训练界面:在平台主界面中,找到并点击目标分类训练功能入口,进入训练界面。界面主要分为设置区和命令区,设置区的参数设置与目标检测训练类似。

  2. 设置训练参数:在设置区,设置各项训练参数,包括迭代次数、批量大小、图像大小、等待轮数、初始学习率、是否保存模型权重、保存模型权重周期、训练设备(和目标检测一样,暂时只支持CPU)、工作线程数、优化器类型等。具体设置方法和参数含义与目标检测训练相同。

  3. 选择模型文件和数据集路径:点击相应的按钮,选择预训练模型文件(一般是models目录下的yolo8n-cls.pt 文件)和数据集路径(要训练的数据分类数据集根目录),选择后文件路径将显示在对应的主界面中。

  4. 开始训练:设置好所有参数后,点击命令区的“训练”按钮,系统会开始训练模型。训练过程和结果显示与目标检测训练相似,训练结束后会在控制台区显示相关的训练指标。

后置过程:在控制台区的窗口的最后两行中找到Results saved to 的目录,在资源管理器打开,并找到其子目录weights,将best.pt拷贝到models目录下,并将文件名改名为数据集的名称,为将来预测做准备


目标分类训练界面代码层级关系

plaintext

- ClassficationTrainingUI (类)
  - __init__ (方法): 类的初始化方法
  - ui_init (方法): 初始化用户界面
  - create_control_area (方法): 创建控制区域
  - create_setting_area (方法): 创建设置区域
  - create_command_area (方法): 创建命令区域
  - select_model (方法): 选择模型
  - select_data (方法): 选择数据
  - start_training (方法): 开始训练
  - train_model (方法): 训练模型
  - get_int_value (方法): 获取整数值
  - get_float_value (方法): 获取浮点数值
  - parse_and_display_output (方法): 解析并显示输出
  - update_console (方法): 更新控制台
  - plot_metrics (方法): 绘制指标图表
- list_sorted_by_mtime (函数): 按修改时间对目录下的文件和目录进行排序并列出

目标分类训练界面功能描述

  1. ClassficationTrainingUI (类)
    :这是一个继承自QWidget
    的类,用于创建一个对象分类训练的用户界面。
    • *init(方法)*:类的初始化方法,通常用于设置对象的初始状态。
    • ui_init (方法)
      :初始化用户界面,可能包括布局设置、控件添加等操作。
    • create_control_area (方法)
      :创建控制区域,可能包含一些控制训练过程的按钮或选项。
    • create_setting_area (方法)
      :创建设置区域,用于设置训练的相关参数。
    • create_command_area (方法)
      :创建命令区域,可能包含执行训练命令的按钮等。
    • select_model (方法)
      :提供选择模型的功能,可能会弹出文件选择对话框让用户选择模型文件。
    • select_data (方法)
      :提供选择数据的功能,可能会弹出文件选择对话框让用户选择训练数据文件。
    • start_training (方法)
      :启动训练过程,可能会调用train_model
      方法。
    • train_model (方法)
      :实际执行模型训练的方法。
    • get_int_value (方法)
      :从输入框中获取整数值,如果输入无效则使用默认值。
    • get_float_value (方法)
      :从输入框中获取浮点数值,如果输入无效则使用默认值。
    • parse_and_display_output (方法)
      :解析训练过程的输出并显示在界面上。
    • update_console (方法)
      :更新控制台的显示内容。
    • plot_metrics (方法)
      :绘制训练过程中的指标图表,如准确率、损失值等。
  2. list_sorted_by_mtime (函数)
    :获取指定目录下所有文件和目录的名称,并按修改时间进行排序。


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

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

评论