一种基于ARM架构的空间目标检测模型构建方法与流程

一种基于arm架构的空间目标检测模型构建方法
技术领域
1.本发明涉及目标检测模型构建技术领域，尤其涉及一种基于arm架构的空间目标检测模型构建方法。


背景技术：

2.arm架构系统平台由处理器、存储等硬件平台与操作系统、应用等软件平台两部分组成，arm架构处理器具备鲁棒性高、实时性好、功耗低和体积小等特点，可用于灵活多变的应用场景。相比于x86架构的计算机系统， arm架构系统执行的任务比较单一，因此可对硬件进行较多简化，降低能耗，更适合嵌入式应用于航天载荷平台。嵌入式架构，常用的如arm架构与常见的x86架构有较大区别，针对运行于x86架构上的空间目标智能检测算法模型，需经过转换、加速处理等系列操作，才能在arm架构设备上高效运行。
3.本发明目的是公开一种把x86架构上的空间目标智能检测算法模型转换为arm架构系统的空间目标检测模型的方法。
4.经检索，并未发现与本技术相同或相近现有技术。


技术实现要素：

5.本发明的目的在于提供一种基于arm架构的空间目标检测模型构建方法，把x86架构上的空间目标智能检测算法模型转换为arm架构系统的空间目标检测模型。
6.本发明的目的可以通过以下技术方案实现：一种基于arm架构的空间目标检测模型构建方法，包括以下步骤：s1、采集空间目标图像，使用目标检测算法对空间目标图像进行训练，获取空间目标图像训练集；s2、以空间目标图像训练集作为输入，经x86架构服务器运算后，获得x86架构下的空间目标检测模型；s3、对x86架构下的空间目标检测模型进行处理，使处理后的空间目标检测模型适用于在arm架构中部署。
7.进一步地：所述使用目标检测算法对空间目标图像进行训练，获取空间目标图像训练集；使用的训练模型为one-stage训练结构模型。
8.进一步地：所述对x86架构下的空间目标检测模型进行处理的步骤包括：s31、对空间目标检测模型进行格式转换；s32、对空间目标检测模型进行量化处理；s33、对空间目标检测模型进行加速处理；s34、对空间目标检测模型进行封装处理；s35、对空间目标检测模型进行arm架构系统适配。
9.进一步地：所述对x86架构下的空间目标检测模型进行处理的步骤还包括：s36、对arm架构系统运行的空间目标检测模型进行智能检测识别验证。
10.进一步地：所述s31中对空间目标检测模型进行格式转换，是将空间目标检测模型转换为onnx格式模型。
11.进一步地：所述s32对空间目标检测模型进行量化处理，是采用8bit量化法对空间目标检测模型进行量化。
12.进一步地：所述s33对空间目标检测模型进行加速处理；是将空间目标检测模型采用float32浮点运算转换为采用float8浮点运算。
13.进一步地：所述s35对空间目标检测模型进行arm架构系统适配，是对空间目标检测模型适配linuxarm架构操作系统。
14.本发明的有益效果：1、本发明通过对x86架构下的空间目标检测模型进行处理，使处理后的空间目标检测模型适用于在arm架构中部署，arm架构的空间目标检测模型，可对硬件进行简化，降低能耗，实时性好、体积小更适合嵌入式的应用于航天载荷平台。
15.2、本发明采用one-stage训练结构模型对空间目标图像进行训练，获取空间目标图像训练集，获取的空间目标图像训练集准确率高，适用于arm架构的空间目标检测模型快速检测识别的要求。
16.3、本发明通过对基于x86架构的空间目标检测模型，经过格式转换、模型量化、模型加速、模型封装、推理参数设置、arm架构系统适配等操作，使空间目标检测模型可部署运行于arm架构系统，实施空间目标检测，处理方法过程简单高效，获取的arm架构系统空间目标检测模型可靠性好，能耗低，硬件要求低，更适合应用于航天载荷平台。
附图说明
17.图1为本发明一种基于arm架构的空间目标检测模型构建方法的流程示意图；图2为本发明arm架构中空间目标检测模型检测分析推理的一种效果图；图3为本发明arm架构中空间目标检测模型检测分析推理的另一种效果图；图4为本发明arm架构中空间目标检测模型检测分析推理的另一种效果图；图5为本发明arm架构中空间目标检测模型检测分析推理的另一种效果图；图6为本发明目标检测算法网络结构示意图。
具体实施方式
18.下面详细描述本发明的实施例，所述实施例的示例在附图中表示，其中自始至终相同或类似的符号表示相同或类似的元件或具有相同或类似功能的元件。下面通过参考附图描述的实施例是示例性的，仅用于解释本发明，而不能理解对本发明的限制。
19.如图1-图6所示，本发明公开一种基于arm架构的空间目标检测模型构建方法，包括以下步骤：s1、采集空间目标图像，使用目标检测算法对空间目标图像进行训练，获取空间目标图像训练集；s2、以空间目标图像训练集作为输入，经x86架构服务器运算后，获得x86架构下的空间目标检测模型；s3、对x86架构下的空间目标检测模型进行处理，使处理后的空间目标检测模型适
用于在arm架构中部署。
20.采集空间目标图像，可以利用公开的或内部收集的空间目标图像，经数据预处理后形成训练集、验证集和测试集，使用目标检测算法对空间目标图像进行训练，训练模型可以采用one-stage训练结构模型。
21.如图6所示，one-stage训练结构模型，分为输入端、backbone、neck和prediction四个部分，输入端采用mosaic、mixup数据增强，backbone上采用了cspdarknet53、mish激活函数、dropblock等方式，neck中采用了spp、fpn的结构，输出端则采用多尺度输出，小尺寸feature map负责检测大目标，大尺寸feature map负责检测小目标。
22.采用one-stage训练结构模型对空间目标图像进行训练，获取空间目标图像训练集，获取的空间目标图像训练集准确率高，适用于arm架构的空间目标检测模型快速检测识别的要求。
23.通过使用目标检测算法对空间目标图像进行训练，获取空间目标图像训练集，把空间目标图像训练集作为输入，优化设置训练参数，经x86架构服务器运算后，得到空间目标检测模型。
24.x86架构的空间目标检测模型一般硬件配置要求高，能耗大，arm架构系统算力有限，不足以支撑空间目标检测模型推理运算，需要对x86架构训练的空间目标检测模型进行处理，使其可以基于arm架构系统部署、运行，进而对空间目标进行检测识别。
25.对x86架构下的空间目标检测模型进行处理的步骤包括：s31、对空间目标检测模型进行格式转换；s32、对空间目标检测模型进行量化处理；s33、对空间目标检测模型进行加速处理；s34、对空间目标检测模型进行封装处理；s35、对空间目标检测模型进行arm架构系统适配。
26.具体的，空间目标检测模型进行格式转换，是基于pytorch架构，对x86架构空间目标检测模型调用torchscript中的转换函数，将空间目标检测模型转为onnx格式模型，为空间目标检测模型后续操作奠定基础。
27.根据量化原理，对空间目标检测模型量化方法可以采用直接经验量化算法，基于重建的方法，基于梯度和损失优化的算法等方法；采用位宽常见的有1bit量化，8bit量化，以及任意bit数量化，由于arm架构系统算力有限，兼顾空间目标检测模型量化精度，宜采用8bit量化法对空间目标检测模型进行量化。
28.经过对空间目标检测模型量化处理后，再将空间目标检测模型在x86架构下采用float32浮点运算转换为float8浮点运算，降低浮点预算方法可以降低对硬件配置的要求；再结合层融合、动态张量内存管理等方法，将模型转换为c++语言可调用的状态，加速空间目标检测模型检测运行速度。
29.再针对空间目标检测模型检测需求及空间目标图像特点，将图像预处理(包括图像分辨率调整、图像通道数调整等)过程、模型载入、模型推理、模型释放、结果后处理等功能封装成动态链接库，便于主程序调用，进行整体的模型封装。
30.对封装后的空间目标检测模型进行arm架构系统适配，优选安装定制版的linuxarm架构操作系统，基于该操作系统编译模型封装、调用源码，对特定架构进行适配，
在定制版inuxarm架构操作系统中编译安装调用模型的依赖库，为空间目标检测模型运行提供支撑。
31.如图2、图3、图4、图5所示，为对arm架构系统运行的空间目标检测模型进行智能检测识别验证，可以通过输入新的空间目标图像，经arm架构操作系统中的空间目标检测模型进行智能检测分析推理后，输出推理结果，如图2、图3所示对陨石检测的检测分析推理，还如图4、图5所示对卫星检测析推理结果，以方框形式标识检出的空间目标，以检验空间目标检测模型在arm架构操作系统中检测分析推理的准确性。
32.以上所述，仅为本发明较佳的具体实施方式，但本发明的保护范围并不局限于此，任何熟悉本技术领域的技术人员在本发明揭露的技术范围内，根据本发明的技术方案及其发明构思加以等同替换或改变，都应涵盖在本发明的保护范围之内。

技术特征：
1.一种基于arm架构的空间目标检测模型构建方法，其特征在于：包括以下步骤：s1、采集空间目标图像，使用目标检测算法对空间目标图像进行训练，获取空间目标图像训练集；s2、以空间目标图像训练集作为输入，经x86架构服务器运算后，获得x86架构下的空间目标检测模型；s3、对x86架构下的空间目标检测模型进行处理，使处理后的空间目标检测模型适用于在arm架构中部署。2.根据权利要求1所述的一种基于arm架构的空间目标检测模型构建方法，其特征在于：所述使用目标检测算法对空间目标图像进行训练，获取空间目标图像训练集；使用的训练模型为one-stage训练结构模型。3.根据权利要求1所述的一种基于arm架构的空间目标检测模型构建方法，其特征在于：所述对x86架构下的空间目标检测模型进行处理的步骤包括：s31、对空间目标检测模型进行格式转换；s32、对空间目标检测模型进行量化处理；s33、对空间目标检测模型进行加速处理；s34、对空间目标检测模型进行封装处理；s35、对空间目标检测模型进行arm架构系统适配。4.根据权利要求3所述的一种基于arm架构的空间目标检测模型构建方法，其特征在于：所述对x86架构下的空间目标检测模型进行处理的步骤还包括：s36、对arm架构系统运行的空间目标检测模型进行智能检测识别验证。5.根据权利要求3所述的一种基于arm架构的空间目标检测模型构建方法，其特征在于：所述s31中对空间目标检测模型进行格式转换，是将空间目标检测模型转换为onnx格式模型。6.根据权利要求3所述的一种基于arm架构的空间目标检测模型构建方法，其特征在于：所述s32中对空间目标检测模型进行量化处理，是采用8bit量化法对空间目标检测模型进行量化。7.根据权利要求3所述的一种基于arm架构的空间目标检测模型构建方法，其特征在于：所述s33中对空间目标检测模型进行加速处理；是将空间目标检测模型采用float32浮点运算转换为采用float8浮点运算。8.根据权利要求3所述的一种基于arm架构的空间目标检测模型构建方法，其特征在于：所述s35中对空间目标检测模型进行arm架构系统适配，是对空间目标检测模型适配linuxarm架构操作系统。

技术总结
本发明公开了一种基于ARM架构的空间目标检测模型构建方法，包括采集空间目标图像，使用目标检测算法对空间目标图像进行训练，获取空间目标图像训练集；以空间目标图像训练集作为输入，经X86架构服务器运算后，获得X86架构下的空间目标检测模型；对X86架构下的空间目标检测模型进行处理，使处理后的空间目标检测模型适用于在ARM架构中部署。本发明通过对X86架构下的空间目标检测模型进行处理，使处理后的空间目标检测模型适用于在ARM架构中部署，ARM架构的空间目标检测模型，可对硬件进行简化，降低能耗，实时性好、体积小更适合嵌入式应用于航天载荷平台。用于航天载荷平台。用于航天载荷平台。


技术研发人员：梁志锋 鱼群 张蕾蕾 张轩 白军辉
受保护的技术使用者：中科星图测控技术股份有限公司
技术研发日：2023.08.02
技术公布日：2023/9/14