[发明专利]一种面向物联网的机器学习容器镜像下载系统及其方法

权利要求书

1.一种面向物联网的机器学习容器镜像下载系统，其特征在于，包括一个主控节点和多个计算节点；

所述主控节点：用于存储、转换机器学习模型，将转换格式的机器学习模型构建为机器学习容器镜像；当构建机器学习容器镜像的镜像信息完成后，向计算节点下发镜像下载指令；

所述计算节点：用于接收所述镜像下载指令，下载机器学习容器镜像，并启动机器学习容器；并接收来自物联网设备收集的数据，向物联网设备返回数据处理结果。

2.根据权利要求1所述的一种面向物联网的机器学习容器镜像下载系统，其特征在于，所述机器学习模型包括计算机视觉和自然语言处理领域的预训练机器学习模型。

3.根据权利要求1所述的一种面向物联网的机器学习容器镜像下载系统，其特征在于，所述主控节点包括控制平面、机器学习模型仓库、机器学习模型转换器、机器学习容器镜像构建器、机器学习容器镜像仓库和调度器；

所述控制平面，用于读取、解析用户设置的机器学习容器部署参数，所述机器学习容器部署参数包括机器学习模型的类型及所述机器学习模型依赖的对应的机器学习计算框架和需要部署机器学习容器的计算节点参数；

所述机器学习模型仓库，用于储存机器学习模型；

所述机器学习模型转换器，用于将所述机器学习模型转换为ONNX格式的机器学习模型；

机器学习容器镜像构建器，用于依据转换为ONNX格式的机器学习模型构建所述机器学习容器镜像；

所述机器学习容器镜像仓库，用于存储带有镜像信息的所述机器学习容器镜像；

所述调度器，用于接收控制平面设置的需要部署所述机器学习容器的计算节点参数和所述机器学习容器镜像的镜像信息，向所述计算节点发送镜像下载指令。

4.根据权利要求3所述的一种面向物联网的机器学习容器镜像下载系统，其特征在于，所述计算节点包括镜像分发代理、容器引擎和分布式存储组件；

所述镜像分发代理，用于接收所述镜像下载指令、通知容器引擎下载所述机器学习容器镜像、拦截容器引擎对所述机器学习容器镜像的下载请求、访问所在计算节点的分布式存储组件；

所述容器引擎，用于接收镜像分发代理的通知，发起下载所述机器学习容器镜像的请求；

所述分布式存储组件，用于存储下载的所述机器学习容器镜像，各分布式存储组件之间相互连接，为所述镜像分发代理获取包含所述机器学习容器镜像的下载源，并从获取的所述下载源处下载所述机器学习容器镜像。

5.根据权利要求4所述的一种面向物联网的机器学习容器镜像下载系统，其特征在于，所述分布式存储组件采用Bitswap协议从机器学习容器镜像的下载源下载机器学习容器镜像。

6.根据权利要求3所述的一种面向物联网的机器学习容器镜像下载系统，其特征在于，构建完成的所述机器学习容器镜像包括：执行所述机器学习模型的运行脚本及其依赖库。

7.一种面向物联网的机器学习容器镜像下载系统的方法，其特征在于，所述方法包括以下步骤：

所述控制平面读取所述用户设置的机器学习容器部署参数，解析出机器学习模型的名称通过机器学习模型仓库查找所需的机器学习模型；所述机器学习容器镜像构建器接收经转换后的所述ONNX格式的机器学习模型，构建为机器学习容器镜像；所述机器学习容器镜像仓库将存储的所述机器学习容器镜像的信息发送至所述调度器；

所述镜像分发代理接收镜像下载指令后，对应所述分布式存储组件从所述机器学习容器镜像的下载源获取所述机器学习容器镜像的文件全部；启动对应的所述机器学习容器；计算节点接收物联网设备收集到的数据，并交由启动的所述机器学习容器进行数据处理；镜像分发代理将所述机器学习容器的数据处理结果反馈至物联网设备。