[發(fā)明專利]地表覆蓋物識別方法、神經(jīng)網(wǎng)絡(luò)的訓(xùn)練方法及裝置

在審

• 申請?zhí)枺?/td>: 202010606948.7
• 申請日: 2020-06-29
• 公開（公告）號: CN111753773A
• 公開（公告）日: 2020-10-09
• 發(fā)明（設(shè)計）人: 董潤敏
• 申請（專利權(quán)）人: 上海商湯智能科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62;G06N3/04
• 代理公司: 北京中知法苑知識產(chǎn)權(quán)代理有限公司 11226
• 代理人: 李明;趙吉陽

1.一種地表覆蓋物識別方法，其特征在于，包括：

獲取地表圖像；

利用目標神經(jīng)網(wǎng)絡(luò)對所述地表圖像進行像素點級別的分類識別，得到所述地表圖像中的像素點所屬的地表覆蓋物的類別信息；

其中，所述目標神經(jīng)網(wǎng)絡(luò)為利用地表樣本圖像以及所述地表樣本圖像中樣本像素點對應(yīng)的目標標簽信息訓(xùn)練得到，所述目標標簽信息用于標識所述樣本像素點所屬的地表覆蓋物類別，所述目標標簽信息為通過對所述地表樣本圖像中樣本像素點對應(yīng)的初始標簽信息進行調(diào)整后得到的。

2.根據(jù)權(quán)利要求1所述的地表覆蓋物識別方法，其特征在于，在得到所述地表圖像中的像素點所屬的地表覆蓋物的類別信息后，所述地表覆蓋物識別方法還包括：

基于地表圖像中每個像素點對應(yīng)的類別信息，確定該像素點在所述地表圖像對應(yīng)的地表覆蓋圖像中的顯示色彩；

基于每個像素點對應(yīng)的顯示色彩，生成所述地表圖像對應(yīng)的地表覆蓋圖像。

3.一種神經(jīng)網(wǎng)絡(luò)的訓(xùn)練方法，其特征在于，包括：

獲取地表樣本圖像以及與所述地表樣本圖像對應(yīng)的初始地表覆蓋圖像，所述初始地表覆蓋圖像中的每個像素點對應(yīng)有指示地表覆蓋物類別的初始標簽信息；

基于所述初始地表覆蓋圖像中的每個像素點對應(yīng)的初始標簽信息和所述地表樣本圖像，確定所述地表樣本圖像中每個像素點對應(yīng)的目標標簽信息；

基于所述地表樣本圖像和第一神經(jīng)網(wǎng)絡(luò)，預(yù)測所述地表樣本圖像中的每個像素點屬于多種預(yù)設(shè)地表覆蓋物類別中每種地表覆蓋物類別的第一概率；

基于每個像素點對應(yīng)的所述第一概率和所述目標標簽信息，對所述第一神經(jīng)網(wǎng)絡(luò)中的網(wǎng)絡(luò)參數(shù)值進行調(diào)整，得到目標神經(jīng)網(wǎng)絡(luò)。

4.根據(jù)權(quán)利要求3所述的訓(xùn)練方法，其特征在于，所述基于所述初始地表覆蓋圖像中的每個像素點對應(yīng)的初始標簽信息和所述地表樣本圖像，確定所述地表樣本圖像中每個像素點對應(yīng)的目標標簽信息，包括：

基于所述地表樣本圖像以及待訓(xùn)練的第二神經(jīng)網(wǎng)絡(luò)，預(yù)測所述地表樣本圖像中每個像素點屬于每種地表覆蓋物類別的第二概率；

基于每個像素點對應(yīng)的所述第二概率和所述初始標簽信息，對所述第二神經(jīng)網(wǎng)絡(luò)中的網(wǎng)絡(luò)參數(shù)值進行調(diào)整，得到初始第二神經(jīng)網(wǎng)絡(luò)；

基于所述地表樣本圖像、所述初始第二神經(jīng)網(wǎng)絡(luò)和每個像素點對應(yīng)的所述初始標簽信息，確定所述地表樣本圖像中每個像素點對應(yīng)的目標標簽信息。

5.根據(jù)權(quán)利要求4所述的訓(xùn)練方法，其特征在于，所述基于所述地表樣本圖像、所述初始第二神經(jīng)網(wǎng)絡(luò)和每個像素點對應(yīng)的所述初始標簽信息，確定所述地表樣本圖像中每個像素點對應(yīng)的目標標簽信息，包括：

將所述地表樣本圖像輸入當前第二神經(jīng)網(wǎng)絡(luò)，得到當前次預(yù)測的所述地表樣本圖像中每個像素點屬于每種地表覆蓋物類別的第二概率；所述當前第二神經(jīng)網(wǎng)絡(luò)為所述初始第二神經(jīng)網(wǎng)絡(luò)，或者為對所述初始第二神經(jīng)網(wǎng)絡(luò)經(jīng)過至少一次調(diào)整后得到的神經(jīng)網(wǎng)絡(luò)；

基于當前次預(yù)測的所述地表樣本圖像中每個像素點屬于每種地表覆蓋物類別的第二概率，確定所述地表樣本圖像中每個像素點對應(yīng)的當前次調(diào)整后的標簽信息；

基于當前次預(yù)測的每個像素點對應(yīng)的所述第二概率、每個像素點對應(yīng)的初始標簽信息、以及每個像素點對應(yīng)的上一次調(diào)整后的標簽信息，對所述當前第二神經(jīng)網(wǎng)絡(luò)中的網(wǎng)絡(luò)參數(shù)值進行調(diào)整，得到調(diào)整后的當前第二神經(jīng)網(wǎng)絡(luò)；

在未滿足預(yù)設(shè)截止條件的情況下，返回將所述地表樣本圖像輸入當前第二神經(jīng)網(wǎng)絡(luò)的步驟，直到滿足預(yù)設(shè)截止條件后，將當前次調(diào)整后的每個像素點的標簽信息作為該像素點對應(yīng)的所述目標標簽信息。

6.根據(jù)權(quán)利要求5所述的訓(xùn)練方法，其特征在于，所述預(yù)設(shè)截止條件包括：

當前調(diào)整次數(shù)達到設(shè)定次數(shù)閾值和當前第二神經(jīng)網(wǎng)絡(luò)對應(yīng)的損失值小于設(shè)定閾值中的至少一種，以及，

所述地表樣本圖像中存在超過設(shè)定個數(shù)的目標像素點；所述目標像素點對應(yīng)的當前次調(diào)整后的標簽信息和對應(yīng)的上一次調(diào)整后的標簽信息相同。