(19)中华 人民共和国 国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202111672590.9 (22)申请日 2021.12.31 (71)申请人 广东三维家信息科技有限公司 地址 510000 广东省广州市天河区天河软 件园软件路15号 （孵化二期F栋） 9楼 902室 (72)发明人 邱辉平　黄志艺　卢燕青　王胜　 (74)专利代理 机构 北京超凡宏宇专利代理事务 所(特殊普通 合伙) 11463 代理人 周宇 (51)Int.Cl. G06T 3/40(2006.01) G06N 3/08(2006.01) G06N 3/04(2006.01) (54)发明名称 一种超分辨率图像生成方法、 装置、 电子设 备及存储介质 (57)摘要 本申请实施例提供一种超分辨率图像生成 方法、 装置、 电子设备及存储介质， 其中， 该方法 包括： 将原始图像输入基于Pytorch框架的超分 辨率神经网络模型进行测试， 得到Pytor ch模型； 将Pytorch模型加载至推理加速引擎； 根据推理 加速引擎处理原始图像， 得到裁剪图像； 根据加 载至推理加速引擎的Pytorch模型处理裁剪图 像， 输出超分辨率图像。 实施本申请实施例， 提高 了超分辨率图像的生成速度和效率， 节省了内存 空间的占用。 权利要求书2页 说明书8页 附图2页 CN 114331847 A 2022.04.12 CN 114331847 A 1.一种超分辨 率图像生成方法， 其特 征在于， 所述方法包括： 将原始图像输入基于P ytorch框架的超分辨率神经网络模型进行测试， 得到P ytorch模 型； 将所述Pytorc h模型加载至推理加速引擎； 根据所述推理加速引擎处 理所述原 始图像， 得到 裁剪图像； 根据所述加载至推理加速引擎的所述Pytorch模型处理所述裁剪图像， 输出超分辨率 图像。 2.根据权利要求1所述的超分辨率 图像生成方法， 其特征在于， 所述将所述Pytorch模 型加载至推理加速引擎的步骤， 包括： 获取所述原始图像的尺寸和批次； 根据所述原 始图像的尺寸和批次获得On nx模型的参数； 根据所述 参数将所述Pytorc h模型转换为对应的On nx模型； 将所述On nx模型加载至所述推理加速引擎。 3.根据权利要求2所述的超分辨率图像生成方法， 其特征在于， 所述将所述Onnx模型加 载至所述推理加速引擎的步骤， 包括： 解析所述Onnx模型， 得到相应的配置文件； 根据所述原 始图像的尺寸和批次设置所述配置文件的参数； 将所述参数结合并行计算框架对所述On nx模型进行序列转换， 完成加载。 4.根据权利要求1所述的超分辨率图像生成方法， 其特征在于， 所述根据 所述推理加速 引擎处理所述原 始图像， 得到 裁剪图像的步骤， 包括： 获取所述推理加速引擎的测试环境； 根据所述测试环境对所述原 始图像进行裁 剪， 得到所述裁 剪图像。 5.根据权利要求4所述的超分辨率图像生成方法， 其特征在于， 所述根据 所述加载至推 理加速引擎的所述Pytorc h模型处理所述裁 剪图像， 输出超分辨 率图像的步骤， 包括： 根据所述测试环境的信息构建所述推理加速引擎的推理环境； 根据所述推理环境加载 所述推理加速引擎； 根据加载后的所述推理加速引擎和所述裁 剪图像输出超分辨 率图像。 6.根据权利要求5所述的超分辨率图像生成方法， 其特征在于， 所述根据 所述测试环境 的信息构建所述推理加速引擎的推理环境的步骤， 包括： 获取所述裁剪图像的尺寸和批次； 获取内存信息、 显存信息和地址信息； 构建多线程， 在所述多线程中根据 所述裁剪图像的尺寸和批次分别将所述内存信 息中 的内存、 所述显存信息中的显存和所述 地址信息中的地址进行分配， 得到推理数据流。 7.根据权利要求5所述的超分辨率图像生成方法， 其特征在于， 在所述根据所述推理加 速引擎和所述裁 剪图像输出超分辨 率图像的步骤之后， 还 包括： 释放所述推理加速引擎 运行过程中所占用的内存及显存空间。 8.一种超分辨 率图像生成装置， 其特 征在于， 所述装置包括： 测试模块， 用于将原始图像输入基于Pytorch框架的超分辨率神经网络模型进行测试， 得到Pytorc h模型；权　利　要　求　书 1/2 页 2 CN 114331847 A 2加载模块， 用于将所述Pytorc h模型加载至推理加速引擎 裁剪模块， 用于根据所述推理加速引擎处 理所述原 始图像， 得到 裁剪图像； 输出模块， 用于根据所述加载至推理加速引擎的所述Pytorch模型处理所述裁剪图像， 输出超分辨 率图像。 9.一种电子设备， 其特征在于， 包括存储器及处理器， 所述存储器用于存储计算机程 序， 所述处理器运行所述计算机程序以使 所述电子 设备执行根据权利要求 1至4中任一项 所 述的超分辨 率图像生成方法。 10.一种计算机可读存储介质， 其特征在于， 其存储有计算机程序， 所述计算机程序被 处理器执行时实现如权利要求1至4中任一项所述的超分辨 率图像生成方法。权　利　要　求　书 2/2 页 3 CN 114331847 A 3