自定义检测模型
视频分析目标检测提供YOLO系列、VisDrone、道路破损、烟雾火灾、十五种模型。
同时也支持使用自定义的检测模型，包括外部别人训练的模型和自定义训练生成的模型。自定义生成模型需要先通过图片样本管理功能得到样本库，然后使用模型训练工具训练样本库得到检测模型。
下面将为您详细介绍相关内容：
操作步骤
准备样本库
- 创建样本库：单击开始选项卡->浏览组->样本管理下拉按钮->图片样本管理按钮，弹出创建/打开样本库对话框，若存在已有样本库，则单击选择样本库按钮，选择一个已有的样本库打开；若需要新建样本库，则单击创建样本库按钮，然后设置以下参数：
- 样本库名称：输入样本库名称；
- 创建样本类型：在下拉选项中选择样本库类型，提供图片分类和图片标注两种类型，这里选择图片标注；
- 原始图片路径：：选择样本图片所在文件路径，支持添加文件夹根目录及子文件夹中的 JPG文件（*.jpg、*.jpeg）、PNG 文件(*.png)、TIFF文件（*.tif，*.tiff）；
- 最小尺寸：设置样本图片最小尺寸的宽度和高度，当图片小于最小尺寸时，则过滤该图片，不将其添加至样本库。单位为像素。
完成以上参数设置之后，单击确定按钮，即可完成样本库的创建。完成样本库创建后，窗口左侧会出现样本库列表，同时样本图片窗口中将显示列表中的第一张图片。
- 图片标注：在样本图片窗口依次标记检测样本，即在所有样本图片中依次分别标记出不同类别的目标物，并为目标物添加对应类别的标注。提供AI标注、面、线、批量绘制四种样本标签绘制方式。选择其中一种合适的方式，对图片中目标物绘制标签。样本标签绘制详细内容可参见图片标注样本管理页面。
- 导出样本库：模型训练需基于样本库进行训练，绘制好样本后，可将样本导出为样本库。单击图片样本管理选项卡->样本管理组->导出样本库按钮，在弹出的对话框中设置以下参数：
- 样本库名称：设置样本库名称，默认为创建时的样本库名称，作为模型训练用途的样本库名称中不能含有中文字符。
- 文件路径：设置样本库的保存路径。
- 导出类型：用于优化导入样本库时的导入速度，数据量特别大的情况下选择携带数据，下次导入该样本库使用时可提升导入速度。
- 样本库用途：提供目标检测、语义分割、实例分割、视频目标检测四种样本库用途，这里选择视频目标检测。
- 图片缩放系数：可对导出的图片进行缩放，默认不缩放。一般情况下保持默认即可。
注意事项：一般情况下样本库中需要几千张图片作为样本数据，以此训练出来的模型精度才越高。
模型训练
- 导出样本库后就可以用于模型训练，单击工具箱->机器学习->视频分析->模型训练工具。
- 在弹出的模型训练对话框中选择样本库文件夹，并设置相关参数，详细参数说明请参见模型训练参数说明。
添加自定义检测模型
- 完成模型训练之后，可将模型添加进来用于视频检测分析。单击视频分析选项卡->交通分析组->检测设置按钮，弹出检测设置对话框。
- 在检测设置对话框中，模型设置下拉菜单中选择自定义，弹出自定义模型管理对话框，然后单击添加按钮，在弹出的添加模型对话框中设置以下参数：
- 模型算法：选择模型训练时使用的相同的算法。
- 模型文件：选择模型文件，选择生成的模型文件夹中weights文件夹下的best.pt文件。
- 类型文件：选择类型文件，选择模型训练所用的样本库文件夹下的与样本库同名的*.txt文件。
- 训练图片大小：指定训练的图片大小，默认为640，图片大小必须是32的倍数。这里与模型训练时指定的图片大小保持一致。
- 模型名称：根据导入的模型自动填充，也可以自定义。
- 模型描述：根据导入的模型自动填充，也可以自定义。
- 或者自定义模型管理对话框中单击导入按钮，直接从本地文件中导入模型文件（.sdm）。
以上，就完成了从样本管理到训练自定义模型以及使用的全过程。
注意事项：使用以上功能时，产品包的路径不能存在中文字符。
相关主题