AI大模型教程
一起来学习

记录halcon【异常检测】的训练过程--上



一、训练前准备

  • 本次训练使用的是halcon自带的深度学习工具MVTec Deep Learning Tool 24.05
  • 数据集准备TrainData文件夹下分别创建OK文件夹,NG文件夹,OK文件夹里面放OK图像,NG文件夹里面分NG文件

二、搭建训练项目

1.创建项目


2.导入训练数据

1.更改标签名称

把默认的标签名称更改成OK,NG

2.导入训练数据

选择TrainData文件夹,导入数据

导入数据时,根据文件夹分类选择对应的类别,然后选择导入并标注图像


3.拆分数据集

可以自行定义训练集,验证集,测试集的比例,默认70%,15%,15%,设定一个随机种子(随意设置)


三、训练数据


1.神经网络选择


2.训练设备选择

有N卡,可以选择GPU训练,显卡驱动最好更新到最新版本,避免训练出错,
无N卡,选择CPU训练,训练的结果时一样的,耗时比较长而已。


3.训练参数

Epoch数量不宜太大,设置过大,会导致超出GPU的内存,导致报错


4.样本增强

可以根据项目的需求,进行训练样本的扩展增强


5.开始训练

最后:接下来的步骤需要等训练结束后再分享记录halcon【异常检测】的训练过程--下

文章来源于互联网:记录halcon【异常检测】的训练过程–上

相关推荐: 小支同学的阿里云大模型Clouder认证(2个入门级):基于百炼平台构建智能体应用+基于PAI ArtLab的AIGC设计

前言        欢迎参与阿里云大模型 Clouder 认证考试!本次认证包含两个核心模块 ——基于百炼平台构建智能体应用与基于 PAI ArtLab 的 AIGC 设计基础,旨在检验你对大模型技术落地与 AIGC 设计的综合实践能力。        当前,…

赞(0)
未经允许不得转载:5bei.cn大模型教程网 » 记录halcon【异常检测】的训练过程--上
分享到: 更多 (0)

AI大模型,我们的未来

小欢软考联系我们