[發(fā)明專利]一種訓(xùn)練集的拆分方法及裝置

有效

• 申請?zhí)枺?/td>: 201910251191.1
• 申請日: 2019-03-29
• 公開（公告）號: CN110084277B
• 公開（公告）日: 2021-03-05
• 發(fā)明（設(shè)計(jì)）人: 王斐
• 申請（專利權(quán)）人: 廣州思德醫(yī)療科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62
• 代理公司: 北京路浩知識產(chǎn)權(quán)代理有限公司 11002
• 代理人: 王慶龍;苗曉靜

本發(fā)明實(shí)施例提供一種訓(xùn)練集的拆分方法及裝置，所述方法包括：獲取用于訓(xùn)練預(yù)設(shè)模型的訓(xùn)練集；所述訓(xùn)練集包括一類域外訓(xùn)練集；所述一類域外訓(xùn)練集是基于原始圖片的拍攝缺陷、與待檢測目標(biāo)部位無關(guān)的拍攝部位確定的訓(xùn)練集；識別所述一類域外訓(xùn)練集中的圖片特征，若判斷獲知所述圖片特征所屬的類別為第一類別，則選取對照圖片對所述第一類別對應(yīng)的圖片進(jìn)行比照；根據(jù)比照結(jié)果拆分所述一類域外訓(xùn)練集。所述裝置執(zhí)行上述方法。本發(fā)明實(shí)施例提供的訓(xùn)練集的拆分方法及裝置，能夠提高訓(xùn)練集拆分的合理性。

技術(shù)領(lǐng)域

本發(fā)明實(shí)施例涉及圖片處理技術(shù)領(lǐng)域，尤其涉及一種訓(xùn)練集的拆分方法及裝置。

背景技術(shù)

膠囊內(nèi)鏡具有無痛、無傷、拍攝圖像信息量大等優(yōu)點(diǎn)，具備廣泛的應(yīng)用價(jià)值。

現(xiàn)有技術(shù)采用人工方式識別通過膠囊內(nèi)鏡拍攝的原始圖片、并對原始圖片進(jìn)行分類，為了更加準(zhǔn)確、高效地識別原始圖片，需要構(gòu)建模型，但是模型在使用之前通常需要進(jìn)行訓(xùn)練，訓(xùn)練過程中的訓(xùn)練集需要進(jìn)行拆分，以便模型能夠更加準(zhǔn)確地進(jìn)行圖片識別，但是，對于訓(xùn)練集的拆分，目前尚沒有有效的方法。

因此，如何避免上述缺陷，提高訓(xùn)練集拆分的合理性，成為亟須解決的問題。

發(fā)明內(nèi)容

針對現(xiàn)有技術(shù)存在的問題，本發(fā)明實(shí)施例提供一種訓(xùn)練集的拆分方法及裝置。

本發(fā)明實(shí)施例提供一種訓(xùn)練集的拆分方法，包括：

獲取用于訓(xùn)練預(yù)設(shè)模型的訓(xùn)練集；所述訓(xùn)練集包括一類域外訓(xùn)練集；所述一類域外訓(xùn)練集是基于原始圖片的拍攝缺陷、與待檢測目標(biāo)部位無關(guān)的拍攝部位確定的訓(xùn)練集；

識別所述一類域外訓(xùn)練集中的圖片特征，若判斷獲知所述圖片特征所屬的類別為第一類別，則選取對照圖片對所述第一類別對應(yīng)的圖片進(jìn)行比照；

根據(jù)比照結(jié)果拆分所述一類域外訓(xùn)練集。

本發(fā)明實(shí)施例提供一種訓(xùn)練集的拆分裝置，包括：

獲取單元，用于獲取用于訓(xùn)練預(yù)設(shè)模型的訓(xùn)練集；所述訓(xùn)練集包括一類域外訓(xùn)練集；所述一類域外訓(xùn)練集是基于原始圖片的拍攝缺陷、與待檢測目標(biāo)部位無關(guān)的拍攝部位確定的訓(xùn)練集；

選取單元，用于識別所述一類域外訓(xùn)練集中的圖片特征，若判斷獲知所述圖片特征所屬的類別為第一類別，則選取對照圖片對所述第一類別對應(yīng)的圖片進(jìn)行比照；

拆分單元，用于根據(jù)比照結(jié)果拆分所述一類域外訓(xùn)練集。

本發(fā)明實(shí)施例提供一種電子設(shè)備，包括：處理器、存儲器和總線，其中，

所述處理器和所述存儲器通過所述總線完成相互間的通信；

所述存儲器存儲有可被所述處理器執(zhí)行的程序指令，所述處理器調(diào)用所述程序指令能夠執(zhí)行如下方法：

獲取用于訓(xùn)練預(yù)設(shè)模型的訓(xùn)練集；所述訓(xùn)練集包括一類域外訓(xùn)練集；所述一類域外訓(xùn)練集是基于原始圖片的拍攝缺陷、與待檢測目標(biāo)部位無關(guān)的拍攝部位確定的訓(xùn)練集；

識別所述一類域外訓(xùn)練集中的圖片特征，若判斷獲知所述圖片特征所屬的類別為第一類別，則選取對照圖片對所述第一類別對應(yīng)的圖片進(jìn)行比照；

根據(jù)比照結(jié)果拆分所述一類域外訓(xùn)練集。

本發(fā)明實(shí)施例提供一種非暫態(tài)計(jì)算機(jī)可讀存儲介質(zhì)，包括：

所述非暫態(tài)計(jì)算機(jī)可讀存儲介質(zhì)存儲計(jì)算機(jī)指令，所述計(jì)算機(jī)指令使所述計(jì)算機(jī)執(zhí)行如下方法：

獲取用于訓(xùn)練預(yù)設(shè)模型的訓(xùn)練集；所述訓(xùn)練集包括一類域外訓(xùn)練集；所述一類域外訓(xùn)練集是基于原始圖片的拍攝缺陷、與待檢測目標(biāo)部位無關(guān)的拍攝部位確定的訓(xùn)練集；