[发明专利]一种训练集的拆分方法及装置

说明书

本发明实施例提供一种训练集的拆分方法及装置，所述方法包括：获取用于训练预设模型的训练集；所述训练集包括一类域外训练集；所述一类域外训练集是基于原始图片的拍摄缺陷、与待检测目标部位无关的拍摄部位确定的训练集；识别所述一类域外训练集中的图片特征，若判断获知所述图片特征所属的类别为第一类别，则选取对照图片对所述第一类别对应的图片进行比照；根据比照结果拆分所述一类域外训练集。所述装置执行上述方法。本发明实施例提供的训练集的拆分方法及装置，能够提高训练集拆分的合理性。

技术领域

本发明实施例涉及图片处理技术领域，尤其涉及一种训练集的拆分方法及装置。

背景技术

胶囊内镜具有无痛、无伤、拍摄图像信息量大等优点，具备广泛的应用价值。

现有技术采用人工方式识别通过胶囊内镜拍摄的原始图片、并对原始图片进行分类，为了更加准确、高效地识别原始图片，需要构建模型，但是模型在使用之前通常需要进行训练，训练过程中的训练集需要进行拆分，以便模型能够更加准确地进行图片识别，但是，对于训练集的拆分，目前尚没有有效的方法。

因此，如何避免上述缺陷，提高训练集拆分的合理性，成为亟须解决的问题。

发明内容

针对现有技术存在的问题，本发明实施例提供一种训练集的拆分方法及装置。

本发明实施例提供一种训练集的拆分方法，包括：

获取用于训练预设模型的训练集；所述训练集包括一类域外训练集；所述一类域外训练集是基于原始图片的拍摄缺陷、与待检测目标部位无关的拍摄部位确定的训练集；

识别所述一类域外训练集中的图片特征，若判断获知所述图片特征所属的类别为第一类别，则选取对照图片对所述第一类别对应的图片进行比照；

根据比照结果拆分所述一类域外训练集。

本发明实施例提供一种训练集的拆分装置，包括：

获取单元，用于获取用于训练预设模型的训练集；所述训练集包括一类域外训练集；所述一类域外训练集是基于原始图片的拍摄缺陷、与待检测目标部位无关的拍摄部位确定的训练集；

选取单元，用于识别所述一类域外训练集中的图片特征，若判断获知所述图片特征所属的类别为第一类别，则选取对照图片对所述第一类别对应的图片进行比照；

拆分单元，用于根据比照结果拆分所述一类域外训练集。

本发明实施例提供一种电子设备，包括：处理器、存储器和总线，其中，

所述处理器和所述存储器通过所述总线完成相互间的通信；

所述存储器存储有可被所述处理器执行的程序指令，所述处理器调用所述程序指令能够执行如下方法：

获取用于训练预设模型的训练集；所述训练集包括一类域外训练集；所述一类域外训练集是基于原始图片的拍摄缺陷、与待检测目标部位无关的拍摄部位确定的训练集；

识别所述一类域外训练集中的图片特征，若判断获知所述图片特征所属的类别为第一类别，则选取对照图片对所述第一类别对应的图片进行比照；

根据比照结果拆分所述一类域外训练集。

本发明实施例提供一种非暂态计算机可读存储介质，包括：

所述非暂态计算机可读存储介质存储计算机指令，所述计算机指令使所述计算机执行如下方法：

获取用于训练预设模型的训练集；所述训练集包括一类域外训练集；所述一类域外训练集是基于原始图片的拍摄缺陷、与待检测目标部位无关的拍摄部位确定的训练集；