[發(fā)明專利]圖像顯示模型的訓(xùn)練方法、裝置、電子設(shè)備和存儲(chǔ)介質(zhì)

有效

• 申請(qǐng)?zhí)枺?/td>: 202110891323.4
• 申請(qǐng)日: 2021-08-04
• 公開(kāi)（公告）號(hào): CN113705620B
• 公開(kāi)（公告）日: 2023-08-15
• 發(fā)明（設(shè)計(jì)）人: 熊俊峰;王洋;張歡;仲震宇;高夢(mèng)晗;呂中厚
• 申請(qǐng)（專利權(quán)）人: 百度在線網(wǎng)絡(luò)技術(shù)（北京）有限公司
• 主分類號(hào): G06V10/774
• 分類號(hào): G06V10/774;G06N20/00
• 代理公司: 北京清亦華知識(shí)產(chǎn)權(quán)代理事務(wù)所(普通合伙) 11201
• 代理人: 張夢(mèng)瑤

1.一種圖像顯示模型的訓(xùn)練方法，包括：

生成樣本圖像；

向顯示設(shè)備發(fā)送所述樣本圖像，以使所述顯示設(shè)備顯示所述樣本圖像；

獲取針對(duì)所述顯示設(shè)備的拍攝圖像；

根據(jù)所述拍攝圖像獲取所述顯示設(shè)備的當(dāng)前姿態(tài)信息；

根據(jù)所述顯示設(shè)備的當(dāng)前姿態(tài)信息判斷是否滿足預(yù)設(shè)條件；

如果判斷滿足所述預(yù)設(shè)條件，則從所述拍攝圖像中提取顯示圖像；

根據(jù)所述顯示圖像及所述樣本圖像中相同位置的像素值之間的差異，對(duì)所述圖像顯示模型進(jìn)行訓(xùn)練，以使所述圖像顯示模型學(xué)習(xí)顏色空間中各個(gè)顏色對(duì)應(yīng)的權(quán)重值；

預(yù)先設(shè)置初始環(huán)境變量補(bǔ)償矩陣，所述環(huán)境變量補(bǔ)償矩陣中包括顏色空間中每個(gè)顏色對(duì)應(yīng)的權(quán)重值；

確定所述每個(gè)顏色對(duì)應(yīng)的權(quán)重值的更新量，并根據(jù)所述更新量對(duì)初始權(quán)重值進(jìn)行更新，直至損失函數(shù)收斂。

2.如權(quán)利要求1所述的方法，其中，所述根據(jù)所述拍攝圖像獲取所述顯示設(shè)備的當(dāng)前姿態(tài)信息，包括：

對(duì)所述拍攝圖像進(jìn)行邊緣檢測(cè)和/或垂直角特征提取處理，以獲取所述當(dāng)前姿態(tài)信息。

3.如權(quán)利要求2所述的方法，其中，所述根據(jù)所述顯示設(shè)備的當(dāng)前姿態(tài)信息判斷是否滿足預(yù)設(shè)條件，包括：

獲取所述顯示設(shè)備在所述拍攝圖像中的坐標(biāo)矩陣與預(yù)設(shè)坐標(biāo)矩陣之間的距離；

在所述距離小于距離閾值的情況下，判斷所述顯示設(shè)備滿足預(yù)設(shè)條件。

4.如權(quán)利要求3所述的方法，其中，所述從所述拍攝圖像中提取顯示圖像，包括：

根據(jù)所述坐標(biāo)矩陣，確定所述顯示設(shè)備在所述拍攝圖像中的區(qū)域范圍；

根據(jù)所述區(qū)域范圍，從所述拍攝圖像中提取所述顯示圖像。

5.如權(quán)利要求1所述的方法，還包括:

如果判斷未滿足所述預(yù)設(shè)條件，則繼續(xù)對(duì)所述顯示設(shè)備進(jìn)行檢測(cè)，直至滿足所述預(yù)設(shè)條件。

6.如權(quán)利要求1所述的方法，其中，所述生成樣本圖像，包括：

在顏色空間進(jìn)行采樣，生成所述樣本圖像。

7.一種圖像顯示模型的訓(xùn)練裝置，包括：

第一生成模塊，用于生成樣本圖像；

發(fā)送模塊，用于向顯示設(shè)備發(fā)送所述樣本圖像，以使所述顯示設(shè)備顯示所述樣本圖像；

判斷模塊，用于對(duì)所述顯示設(shè)備進(jìn)行檢測(cè)，判斷是否滿足預(yù)設(shè)條件；

所述判斷模塊，包括：

第一獲取單元，用于獲取針對(duì)所述顯示設(shè)備的拍攝圖像；

第二獲取單元，用于根據(jù)所述拍攝圖像獲取所述顯示設(shè)備的當(dāng)前姿態(tài)信息；

判斷單元，用于根據(jù)所述顯示設(shè)備的當(dāng)前姿態(tài)信息判斷是否滿足預(yù)設(shè)條件；

第二生成模塊，用于如果判斷滿足所述預(yù)設(shè)條件，則從所述拍攝圖像中提取顯示圖像；

訓(xùn)練模塊，用于根據(jù)所述顯示圖像和所述樣本圖像對(duì)所述圖像顯示模型進(jìn)行訓(xùn)練；

所述訓(xùn)練模塊，用于：

根據(jù)所述顯示圖像及所述樣本圖像中相同位置的像素值之間的差異，對(duì)所述圖像顯示模型進(jìn)行訓(xùn)練，以使所述圖像顯示模型學(xué)習(xí)顏色空間中各個(gè)顏色對(duì)應(yīng)的權(quán)重值；

預(yù)先設(shè)置初始環(huán)境變量補(bǔ)償矩陣，所述環(huán)境變量補(bǔ)償矩陣中包括顏色空間中每個(gè)顏色對(duì)應(yīng)的權(quán)重值；

確定所述每個(gè)顏色對(duì)應(yīng)的權(quán)重值的更新量，并根據(jù)所述更新量對(duì)初始權(quán)重值進(jìn)行更新，直至損失函數(shù)收斂。

8.如權(quán)利要求7所述的裝置，其中，所述第二獲取單元，用于：

對(duì)所述拍攝圖像進(jìn)行邊緣檢測(cè)和/或垂直角特征提取處理，以獲取所述當(dāng)前姿態(tài)信息。

9.如權(quán)利要求8所述的裝置，其中，所述判斷單元，用于：

獲取所述顯示設(shè)備在所述拍攝圖像中的坐標(biāo)矩陣與預(yù)設(shè)坐標(biāo)矩陣之間的距離；

在所述距離小于距離閾值的情況下，判斷所述顯示設(shè)備滿足預(yù)設(shè)條件。