基于ONNX的目标检测模型库系统构建方法
Abstract:
本发明公开一种基于ONNX的目标检测模型库系统构建方法,根据图像数据集,采用适应的目标检测模型训练,得到训练好的模型;根据实际模型输入,将训练好模型转化为开放神经网络交换ONNX格式,并将模型统一部署到服务器ONNXRuntime环境,实现跨框架的深度学习模型部署;使用ONNXRuntime得到模型推理结果,为每个模型编写独立的推理代码,将结果转化为系统所需的统一格式;根据检测结果进行可视化处理,在系统中展示检测后的图像以及相应的评价指标;模型库系统包含模型注册、模型展示和模型服务三个模块。模型注册是将ONNX格式模型上传到系统中;模型展示是展现模型库系统中所有已注册模型;模型服务是用户根据需求调用模型在ONNXRuntiem环境下推理,并展示推理结果。
Public/Granted literature
Patent Agency Ranking
0/0