[發明專利]數據集生成方法及裝置、電子設備和存儲介質

在審

• 申請號: 202110763443.6
• 申請日: 2021-07-06
• 公開（公告）號: CN113505100A
• 公開（公告）日: 2021-10-15
• 發明（設計）人: 張軍偉;李誠
• 申請（專利權）人: 上海商湯智能科技有限公司
• 主分類號: G06F16/16
• 分類號: G06F16/16;G06F16/51;G06K9/62
• 代理公司: 北京林達劉知識產權代理事務所(普通合伙) 11277
• 代理人: 劉新宇

本公開涉及一種數據集生成方法及裝置、電子設備和存儲介質，所述方法包括：將智能設備采集的圖像數據存儲于預設文件夾中，得到目標數據集；響應于針對目標數據集的劃分操作，按照預設的劃分規則，劃分目標數據集為多個子數據集，多個子數據集用于實現針對圖像處理網絡多個訓練階段的訓練。本公開實施例可實現更加便捷地實現針對圖像處理網絡的多個訓練階段的訓練。

技術領域

本公開涉及計算機技術領域，尤其涉及一種數據集生成方法及裝置、電子設備和存儲介質。

背景技術

隨著科學技術的發展，越來越多的機器人都具有拍攝功能，機器人可以拍攝圖像并進行本地存儲，以便于利用拍攝的不同圖像進行神經網絡的網絡訓練。但是現有的機器人對拍攝的圖像進行存儲時，都是將所有拍攝的圖像直接存儲于一個文件夾里，沒有對拍攝的圖像進行管理，這也不便于利用拍攝的圖像實現神經網絡不同訓練階段的訓練。

發明內容

本公開提出了一種數據集生成技術方案。

根據本公開的一方面，提供了一種數據集生成方法，包括：將智能設備采集的圖像數據存儲于預設文件夾中，得到目標數據集；響應于針對所述目標數據集的劃分操作，按照預設的劃分規則，劃分所述目標數據集為多個子數據集，所述多個子數據集用于實現針對圖像處理網絡多個訓練階段的訓練。

在一種可能的實現方式中，所述方法還包括：響應于針對文件夾的創建操作，按照所述創建操作指示的存儲路徑及文件夾名稱，創建文件夾；其中，將智能設備采集的圖像數據存儲于預設文件夾中，得到目標數據集，包括：響應于針對已創建的文件夾的選擇操作，將選中的文件夾確定為所述預設文件夾；將所述智能設備采集的圖像數據，按照所述預設文件夾對應的存儲路徑，存儲于所述預設文件夾中，得到所述目標數據集。

在一種可能的實現方式中，所述劃分操作包括：選擇待劃分的目標數據集和/或設置劃分規則，所述劃分規則包括劃分數量與劃分比例；其中，響應于針對所述目標數據集的劃分操作，按照預設的劃分規則，劃分所述目標數據集為多個子數據集，包括：按照所述劃分規則中的劃分數量，創建與所述劃分數量對應的多個文件夾，各個文件夾的文件夾名稱包括：所述預設文件夾的文件夾名稱以及預設后綴，所述預設后綴與所述圖像處理網絡的各個訓練階段對應；按照所述劃分規則中的劃分比例以及所述多個文件夾的文件夾名稱，將所述目標數據集中的圖像數據分配至所述多個文件夾中，得到與圖像處理網絡的各個訓練階段分別對應的各個子數據集。

在一種可能的實現方式中，所述目標數據集包括多個，所述方法還包括：響應于針對多個目標數據集的合并操作，根據所述合并操作指示的目標文件夾以及選中的多個目標數據集，將所述選中的多個目標數據集中的圖像數據復制到所述目標文件夾中，得到合并后的目標數據集。

在一種可能的實現方式中，所述智能設備與云端服務器之間存在通信連接，所述方法還包括：響應于針對數據集或數據集中圖像數據的同步操作，將所述同步操作指示的數據集或圖像數據，同步至所述云端服務器的指定存儲空間中，其中，所述數據集包括：所述目標數據集及所述子數據集中任意一種。

在一種可能的實現方式中，所述智能設備與云端服務器之間存在通信連接，所述方法還包括：在所述目標數據集或所述子數據集中存在新增的圖像數據的情況下，將所述新增的圖像數據同步至所述云端服務器的指定存儲空間中。

在一種可能的實現方式中，所述方法還包括：響應于針對智能設備的圖像采集功能的觸發操作，得到智能設備上設置的圖像采集部件采集的圖像數據；和/或，響應于針對數據集或數據集中圖像數據的查詢操作，顯示查詢到的數據集或顯示查詢到的圖像數據，其中，所述數據集包括所述目標數據集及所述子數據集中的任意一種；和/或，響應于針對數據集中圖像數據的刪除操作，從所述數據集中刪除選中的圖像數據，所述刪除操作包括單個刪除或批量刪除。

在一種可能的實現方式中，所述方法通過Python編程語言，實現所述數據集生成方法對應的操作界面，所述操作界面用于觸發實現所述數據集生成方法的各種操作。