[發(fā)明專利]圖像預測方法、裝置、計算機設備和存儲介質(zhì)

在審

• 申請?zhí)枺?/td>: 202011155152.0
• 申請日: 2020-10-26
• 公開（公告）號: CN112348012A
• 公開（公告）日: 2021-02-09
• 發(fā)明（設計）人: 姚廣
• 申請（專利權）人: 上海眼控科技股份有限公司
• 主分類號: G06K9/32
• 分類號: G06K9/32;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 廣州華進聯(lián)合專利商標代理有限公司 44224
• 代理人: 杜娟娟

1.一種圖像預測方法，其特征在于，所述方法包括：

獲取待測圖像；所述待測圖像中包括多個圖像區(qū)域，且不同的圖像區(qū)域?qū)煌姆诸愵悇e；

將所述待測圖像輸入預設的圖像預測模型，得到預測圖像；其中，所述圖像預測模型為根據(jù)預設的損失函數(shù)訓練得到的，所述預設的損失函數(shù)包括第一損失函數(shù)和第二損失函數(shù)，所述第一損失函數(shù)的值為根據(jù)標準圖像和樣本預測圖像所得到的，所述第二損失函數(shù)的值為根據(jù)所述樣本預測圖像中各圖像區(qū)域所屬的樣本類別和所述標準圖像中各圖像區(qū)域所屬的標準類別所得到的。

2.根據(jù)權利要求1所述的方法，其特征在于，所述圖像預測模型的訓練過程包括：

將樣本圖像輸入預設的待訓練圖像預測模型，得到所述樣本預測圖像；

根據(jù)所述標準圖像和所述樣本預測圖像，得到所述第一損失函數(shù)的值；

獲取所述標準圖像中各圖像區(qū)域所屬的標準類別和所述樣本預測圖像中各圖像區(qū)域所屬的樣本類別；

根據(jù)所述標準圖像中各圖像區(qū)域所屬的標準類別和所述樣本預測圖像中各圖像區(qū)域所屬的樣本類別，得到所述第二損失函數(shù)的值；

根據(jù)所述第一損失函數(shù)的值和所述第二損失函數(shù)的值，得到所述待訓練圖像預測模型的損失函數(shù)的值，對所述待訓練圖像預測模型進行訓練，得到所述圖像預測模型。

3.根據(jù)權利要求2所述的方法，其特征在于，所述獲取所述標準圖像中各圖像區(qū)域所屬的標準類別和所述樣本預測圖像中各圖像區(qū)域所屬的樣本類別，包括：

根據(jù)預設的轉(zhuǎn)換公式，將所述標準圖像中各圖像區(qū)域的像素值的值域轉(zhuǎn)換為對應的標準概率值，根據(jù)所述標準概率值得到所述標準圖像中各圖像區(qū)域所屬的標準類別；

根據(jù)所述預設的轉(zhuǎn)換公式，將所述樣本預測圖像中各圖像區(qū)域的像素值的值域轉(zhuǎn)換為對應的樣本概率值，根據(jù)所述樣本概率值得到所述樣本預測圖像中各圖像區(qū)域所屬的樣本類別。

4.根據(jù)權利要求2所述的方法，其特征在于，所述根據(jù)所述第一損失函數(shù)的值和所述第二損失函數(shù)的值，得到所述待訓練圖像預測模型的損失函數(shù)的值，對所述待訓練圖像預測模型進行訓練，得到所述圖像預測模型，包括：

對所述第一損失函數(shù)的值和所述第二損失函數(shù)的值進行加權求和，得到所述待訓練圖像預測模型的損失函數(shù)的值；

根據(jù)所述待訓練圖像預測模型的損失函數(shù)的值，對所述待訓練圖像預測模型進行訓練，得到所述圖像預測模型。

5.根據(jù)權利要求1-4任一項所述的方法，其特征在于，所述第二損失函數(shù)包括FocalLoss函數(shù)。

6.根據(jù)權利要求1-4任一項所述的方法，其特征在于，所述第一損失函數(shù)包括BCELoss函數(shù)。

7.根據(jù)權利要求3所述的方法，其特征在于，所述預設的轉(zhuǎn)換公式為：式中，Rate表示概率值，sigmoid表示sigmoid函數(shù)，r為參數(shù)，x為所述標準圖像或所述樣本預測圖像中各圖像區(qū)域的像素值，threshold的取值為所述標準圖像或所述樣本預測圖像中各圖像區(qū)域的值階的中心點位置的值，range的取值為所述標準圖像或所述樣本預測圖像中各圖像區(qū)域的值階范圍值的一半。

8.一種圖像預測裝置，其特征在于，所述裝置包括：

第一獲取模塊，用于獲取待測圖像；所述待測圖像中包括多個圖像區(qū)域，且不同的圖像區(qū)域?qū)煌姆诸愵悇e；

預測模塊，用于將所述待測圖像輸入預設的圖像預測模型，得到預測圖像；其中，所述圖像預測模型為根據(jù)預設的損失函數(shù)訓練得到的，所述預設的損失函數(shù)包括第一損失函數(shù)和第二損失函數(shù)，所述第一損失函數(shù)的值為根據(jù)標準圖像和樣本預測圖像所得到的，所述第二損失函數(shù)的值為根據(jù)所述樣本預測圖像中各圖像區(qū)域所屬的樣本類別和所述標準圖像中各圖像區(qū)域所屬的標準類別所得到的。

9.一種計算機設備，包括存儲器和處理器，所述存儲器存儲有計算機程序，其特征在于，所述處理器執(zhí)行所述計算機程序時實現(xiàn)權利要求1至7中任一項所述的方法的步驟。

10.一種計算機可讀存儲介質(zhì)，其上存儲有計算機程序，其特征在于，所述計算機程序被處理器執(zhí)行時實現(xiàn)權利要求1至7中任一項所述的方法的步驟。