项目介绍前言
随着科技的进步和大数据时代的到来,计算机视觉技术逐渐渗透到各个行业领域,为人们的生活和工作带来了诸多便利。近年来,电信诈骗案件频发,给社会带来了巨大的经济损失和心理负担。电诈园区作为电信诈骗的高发区域,其人员和房屋情况复杂,给安全管理带来了极大的挑战。传统的监控手段往往依赖于人工巡查效率低下且容易错过关键信息。因此,利用无人机搭载计算机视觉技术对电诈园区的人员及房屋情况进行智能识别和分析,成为了解决这一问题的有效途径。
实验假设:电诈园区内电诈人员人数10人,关押人质10人,分别入住在园区内不同房屋中;本实验采用稻草人偶和纸质房屋来代替真实环境,用统一红色人偶来代替电诈人员特征信息,统一的蓝色人偶来代替人质特征信息,红色房屋代替电诈人员居住的房屋特征信息,蓝色房屋代替人质人员居住的房屋特征信息。
实验用具:阿木实验室自研p600系列无人机,搭载SpireCV视觉开发套件。
实验步骤:
1.前期数据采集——>2.数据标注及预处理——>3.模型训练——>
4.模型转化及部署——>5.真机实验及结果展示
项目解析及效果展示
- 前期数据采集
前期对待识别目标进行特征数据进行采集训练,及采集红色、蓝色木偶人和红色、蓝色房屋进行图像采集,本次实验大约采集1000余张特征数据图片,并进行编号。
图1 采集图片数据展示
- 数据标注及预处理
接下来,采用SpireCV配套数据标注软件SpireView来进行数据标注,标注格式为yolo格式,具体操作方法请参考阿木实验室官网——SpireCV文档 - 自定义数据集训练检测器 (amovlab.com)
图2 图片数据标注
全部标注完成后,Ctrl+O选择标注格式并输出结果。此处我们选择Yolo detection format,点击OK在保存路径下生成YOLO格式训练文件;对于YOLO训练来说,我们需要scaled_images文件夹、Yolo_labels文件夹和Yolo_categories.names文件.
- 模型训练
我们根据阿木实验室官网——SpireCV文档 - 自定义数据集训练检测器 (amovlab.com)的模型训练方法将标注好的图片和标签数据存放到指定文件夹下,然后打开终端下载所需的YOLO模型,建立YOLO类别名参数文件夹,最后进行模型,具体训练步骤请参照上述给出阿木实验室官网教程。
图3 开始模型训练截图
图4 训练结果混淆矩阵图
图5 F1_curve图
由训练结果可以看出模型综合准确率为98%,模型训练的结果具有较高的准确性。
- 模型转化及部署
在训练完成后,我们将训练结果最好的.pt模型转换为.wts模型,在将.wts模型部署到SpireCV机载计算机上,并将.wts模型转换为.engine模型,最后针对自定义数据集,需要修改~/SpireCV/sv_algorithm_params.json文件,例如,自定义数据集名称为DatasetName,该名称要与权重文件名称对应。
- 真机实验及结果展示