[发明专利]深度学习平台中图片格式的转换方法及装置

说明书

本发明提供一种深度学习平台中图片格式的转换方法，包括：获取待训练图片的图片格式；判断所述待训练图片的图片格式是否为当前训练框架支持的图片格式；若所述待训练图片的图片格式不是当前训练框架支持的图片格式，则对所述待训练图片的图片格式进行转换，生成与所述待训练图片对应的格式转换图片，并将所述格式转换图片输入训练框架进行训练。本发明实施例提供的深度学习平台中图片格式的转换方法中，通过对图片格式不属于当前训练框架支持的图片格式的待训练图片进行图片格式的转换，经转换后的图片格式为当前训练框架支持的图片格式。故无论学习平台当前使用何种训练框架进行训练，用户可选择训练的图片格式可以不受限制，用户体验良好。

技术领域

本发明涉及机器学习技术领域，特别是涉及一种深度学习平台中图片格式的转换方法及装置。

背景技术

随着人工智能技术的发展，机器学习逐渐成为了人们热议的话题，深度学习正是机器学习领域中一个重要的研究方向。深度学习是学习样本数据的内在规律和表示层次，这些学习过程中获得的信息对诸如文字，图片和声音等数据的解释有很大的帮助。

随着科学技术的发展，出现了越来越多实现深度学习目标的深度学习平台，其中，用于图片分析的深度学习平台为常见的深度学习平台之一。在用于图片分析的深度学习平台中，前端主要是开发用户所能接触的前端展示界面，如用户选择待分析的图片进行输入的页面，而后端主要是开发用户不可见的实现逻辑功能等功能的模块，如为实现训练所搭建的训练框架。

在深度学习平台的使用中，前端展示界面与用户体验关联密切。而在现有技术中，用户输入页面所允许输入图片的格式，通常都是有限的。若用户在界面中选择图片的格式不属于该学习平台所允许输入的图片格式，前端展示界面一般会提示用户选择的格式不正确，不支持上传，用户还需自行进行图片处理才能完成输入，这使得用户的使用体验较差。

发明内容

有鉴于此，本发明实施例提供了一种深度学习平台中图片格式的转换方法，以解决前端界面输入图片的格式限制导致用户体验较差的问题。

本发明实施例还提供了一种深度学习平台中图片格式的转换装置，用以保证上述方法实际中的实现及应用。

为实现上述目的，本发明实施例提供如下技术方案：

一种深度学习平台中图片格式的转换方法，包括：

获取待训练图片的图片格式；

判断所述待训练图片的图片格式是否为当前训练框架支持的图片格式；

若所述待训练图片的图片格式不是所述当前训练框架支持的图片格式，则对所述待训练图片的图片格式进行转换，生成与所述待训练图片对应的格式转换图片，并将所述格式转换图片输入训练框架进行训练；所述格式转换图片的图片格式为所述当前训练框架支持的图片格式。

上述的方法，可选的，所述获取待训练图片的图片格式，包括：

解析所述待训练图片的图片信息，获取所述图片信息中包含的所述待训练图片的后缀名；

将所述待训练图片的后缀名，作为所述待训练图片的图片格式。

上述的方法，可选的，所述判断所述待训练图片的图片格式是否为当前训练框架支持的图片格式，包括：

获取所述当前训练框架支持的图片的后缀名；

判断所述待训练图片的后缀名是否属于所述当前训练框架支持的图片的后缀名；

若所述待训练图片的后缀名不属于所述当前训练框架支持的图片的后缀名，则判断所述待训练图片的图片格式不是所述当前训练框架支持的图片格式。

上述的方法，可选的，所述对所述待训练图片的图片格式进行转换，生成与所述待训练图片对应的格式转换图片，包括：