halcon 深度学习标注_Halcon deep learning之目标检测笔记（一）

た入场券 2023-01-08 06:28 459阅读 0赞

设置输入输出的路径

输入路径有两个，一个是图像路径HalconImageDir，一个是Json文件路径PillBagJsonFile。这两个参数用于后续在图像路径中读取coco文件。

输出路径首先定义一个新的文件夹路径，用于存放样本和模型文件等数据。

设置基于深度学习的目标检测模型的路径，该模型文件以.hdl结尾。

设置样本数据文件夹ExampleDataDir：'detect\_pills\_data'

数据字典文件夹路径，用于存放DLSampple文件。

设置存储预处理参数文件的路径。

设置分割数据集参数

分割数据集是为了区分出训练数据集和验证数据集。这一步设置分割的比例，例如：

TrainingPercent := 70

ValidationPercent := 15

则表示训练数据集占70%，验证数据集占15%，剩下的为测试数据集。

同时指定随机数，用于选择随机图片组成数据集。因为想要在复现训练结果时数据集本身是固定的，因此先使用随机数将数据分散，然后进行分割。

创建DLDataset数据集并调整模型

先检查样本数据集文件夹是否存在，不存在就创建一个。

创建数据集DLDataset调用了自定义的函数read\_dl\_dataset\_from\_coco。其输入参数为coco文件名(以.json结尾)和图像路径，输出数据集名为DLDataset。

该函数主要是根据.json文件中的关键字对图像进行解析，如'categories'、'images'、'annotations'。'annotations'里面再读取'bbox'属性。然后将这些属性读取并转化，以key的形式存入数据字典DLDataset和DLSample。

存入数据字典DLDataset的属性有:

 'image\_dir'

 'class\_ids'

 'class\_names'

存入DLSample数据字典的属性有：

 'image\_id'、

 'image\_file\_name'、

 'bbox\_row1'、

 'bbox\_col1'、

 'bbox\_row2'、

 'bbox\_col2'、

 'bbox\_label\_id'、

 'coco\_raw\_annotations'

最后将DLSample数据字典以‘sample’的形式存入DLDataset数据字典。

创建好数据字典之后，读取DLDataset中的'class\_ids'属性，将其设置为深度学习网络模型DLModelHandle的'class\_ids'属性，同时根据网络模型的句柄DLModelHandle将该模型保存到预设的文件夹路径下，完成了目标检测模型的初始化操作。

分割数据集。

这一部分将数据集分割为训练数据集、验证数据集和测试数据集三类。先设置系统随机数SeedRand。再按照之前设置的分割比例，进行数据集的分割。Halcon的例程中使用了自定义函数split\_dl\_dataset完成这一步操作。该函数先从DLDataset中获取样本数据，将样本与标注的类对应起来。然后创建了数据字典，基于单个分类进行分割。在每个类别的图像中提取一定比例用于构建三类数据集。

数据集预处理

先从模型中获取预处理参数，例程中使用了自定义函数create\_dl\_preprocess\_param\_from\_model从深度学习网络中提取预处理参数，如：

'type'

'image\_width'

'image\_height'

'image\_num\_channels'

'image\_range\_min'

'image\_range\_max'

'ignore\_class\_ids'

并创建预处理参数字典DLPreprocessParam。

接下来这部分是最耗时的，即数据的预处理。例程中使用了自定义的preprocess\_dl\_dataset算子，用于对数据字典中的DLDataset、DLPreprocessParam样本进行预处理。

获取DLDataset中的'samples'，存为DLDatasetSamples。这里面DLDatasetSamples一共有400个样本。将每个样本保存为一个数据字典，并对样本进行预处理。处理的步骤包括给每一个DLSamples样本创建数据字典，获取样本的图像，从图像中获取标注框区域，存入DLSamples数据字典中等。

预览预处理数据集

在开始训练之前，建议检查一遍预处理的数据集。使用get\_dict\_tuple算子从数据字典DLDataset中提取样本DatasetSamples。接着对其进行排序。读取排序后的样本并使用dev\_display\_dl\_data算子将其显示出来。