[發明專利]環境檢測模型的生成方法、計算機設備及可讀存儲介質

在審

• 申請號: 201910477777.X
• 申請日: 2019-06-03
• 公開（公告）號: CN110321928A
• 公開（公告）日: 2019-10-11
• 發明（設計）人: 羅英群;歐陽一村;邢軍華;曾志輝;賀濤;許文龍
• 申請（專利權）人: 深圳中興網信科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62
• 代理公司: 北京友聯知識產權代理事務所(普通合伙) 11343
• 代理人: 尚志峰;汪海屏

1.一種環境檢測模型的生成方法，其特征在于，包括：

接收樣本圖片，對所述樣本圖片中的樣本進行分類；

分別統計屬于不同分類的所述樣本的數量，根據所述樣本的數量確定所述樣本中的小樣本；

根據所述小樣本利用生成對抗網絡模型生成補充樣本，將所述補充樣本補充到所述小樣本；

以所述樣本與所述補充樣本組成訓練集，以所述訓練集訓練待訓練模型，生成所述環境檢測模型。

2.根據權利要求1所述的環境檢測模型的生成方法，其特征在于，所述接收樣本圖片，對所述樣本圖片中的樣本進行分類的步驟，具體包括：

接收所述樣本圖片；

提取所述樣本圖片中的樣本；

根據所述樣本的是否污染環境的狀態，對所述樣本進行分類。

3.根據權利要求2所述的環境檢測模型的生成方法，其特征在于，所述分別統計所述樣本圖片中不同樣本的數量，確定所述樣本中數量低于預設閾值的小樣本的步驟，具體包括：

對每個分類的所述樣本的數量進行統計；

確定數量低于預設閾值的所述樣本為所述小樣本。

4.根據權利要求1至3中任一項所述的環境檢測模型的生成方法，其特征在于，所述根據所述小樣本利用所述生成對抗網絡模型生成補充樣本，將所述補充樣本補充到所述小樣本的步驟，具體包括：

根據所述小樣本利用所述生成對抗網絡模型生成預設數量的所述補充樣本；

將所述預設數量的所述補充樣本添加到所述小樣本的分類中。

5.根據權利要求1至3中任一項所述的環境檢測模型的生成方法，其特征在于，在所述根據所述小樣本利用所述生成對抗網絡模型生成補充樣本，將所述補充樣本補充到所述小樣本后，還包括：

統計所述補充樣本與所述小樣本的數量；

判斷所述補充樣本與所述小樣本的總數量是否使所述訓練集達到平衡；

在所述補充樣本與所述小樣本的數量使所述訓練集達到平衡的情況下，進入所述訓練待訓練模型的步驟。

6.根據權利要求5所述的環境檢測模型的生成方法，其特征在于，

在所述補充樣本與所述小樣本的數量未使所述訓練集達到平衡的情況下，進入所述利用所述生成對抗網絡模型生成補充樣本的步驟。

7.根據權利要求1至3中任一項所述的環境檢測模型的生成方法，其特征在于，所述以所述訓練集訓練待訓練模型，生成所述環境檢測模型的步驟，具體包括：

根據所述訓練集訓練所述待訓練模型，驗證所述待訓練模型的訓練效果，根據所述訓練效果調整所述待訓練模型的參數，得到所述環境檢測模型。

8.根據權利要求1至3中任一項所述的環境檢測模型的生成方法，其特征在于，還包括：

對所述環境檢測模型進行測試。

9.一種計算機設備，其特征在于，包括：

存儲器，用于存儲計算機程序；處理器，用于執行所述計算機程序以實現如權利要求1至8中任一項所述的環境檢測模型的生成方法。

10.一種計算機可讀存儲介質，其上存儲有計算機程序，其特征在于，所述計算機程序被處理器執行時實現如權利要求1至8中任一項所述的環境檢測模型的生成方法。