[發明專利]一種圖像標注方法、裝置及電子設備

在審

• 申請號: 202111401273.3
• 申請日: 2021-11-24
• 公開（公告）號: CN114119780A
• 公開（公告）日: 2022-03-01
• 發明（設計）人: 王衛芳;王闖闖;胡正;朱毅博;龔國基
• 申請（專利權）人: 奧比中光科技集團股份有限公司
• 主分類號: G06T7/90
• 分類號: G06T7/90;G06T7/50;G06K9/62;G06F16/16;G06V10/774;G06V10/56
• 代理公司: 深圳漢世知識產權代理事務所(特殊普通合伙) 44578
• 代理人: 冷仔

1.一種圖像標注方法，其特征在于，包括：

獲取對齊的彩色圖像和深度圖像；

利用預設的經訓練的目標檢測模型對彩色圖像進行檢測，獲得所述彩色圖像對應的標簽信息和關鍵點坐標信息，其中，所述標簽信息為所述目標的類別信息；

根據所述彩色圖像對應的標簽信息和關鍵點坐標信息并基于對齊關系生成所述深度圖像對應的標注文件；

將所述深度圖像轉換為偽彩色圖像，并基于所述深度圖像的偽彩色圖像對所述深度圖像對應的標注文件進行修正，得到修正后的標注數據集。

2.如權利要求1所述的圖像標注方法，其特征在于，所述將所述深度圖像轉換為偽彩色圖像，包括：

獲取色度圖，所述色度圖包括顏色值與像素值的映射關系；

對所述深度圖像進行歸一化，得到所述深度圖像對應的歸一化圖像；

根據所述色度圖將歸一化圖像映射成偽彩色圖像。

3.如權利要求1所述的圖像標注方法，其特征在于，所述獲取對齊的彩色圖像和深度圖像，包括：

將所述彩色圖像的像素所述深度圖像的像素進行一一對齊。

4.如權利要求3所述的圖像標注方法，其特征在于，所述根據所述彩色圖像對應的標簽信息和關鍵點坐標信息并基于對齊關系生成所述深度圖像對應的標注文件，包括：

基于所述彩色圖像與深度圖像的對齊關系，將所述彩色圖像對應的標簽信息和關鍵點坐標信息轉換為深度圖像的標注信息，并寫入預設格式的標注文件得到所述深度圖像對應的標注文件。

5.如權利要求4所述的圖像標注方法，其特征在于，所述標注文件還包括深度圖像的圖像信息，其中，所述圖像信息包括圖像的長、寬、通道、路徑和圖像名稱。

6.如權利要求1所述的圖像標注方法，其特征在于，所述修正后的標注數據集包括：

利用所述偽彩色圖像對所述標注文件中包括的所述標簽信息和所述坐標信息進行一一校對，獲得修正后的標注數據集。

7.如權利要求6所述的圖像標注方法，其特征在于，所述利用所述偽彩色圖像對所述標注文件中包括的所述標簽信息和所述坐標信息進行一一校對，包括：

利用標注工具對所述標注文件進行修正，獲取修正后的標注數據集，其中，所述標注工具通過判斷在所述偽彩色圖像上顯示的所述標注文件中包括的標簽信息與坐標信息是否正確對所述標注文件進行修正。

8.一種圖像標注裝置，其特征在于，包括：

采集模塊，用于采集彩色圖像和深度圖像；

對齊模塊，用于將對齊所述彩色圖像和所述深度圖像；

檢測模塊，用于利用預設的經訓練的目標檢測模型對彩色圖像進行檢測，獲得所述彩色圖像對應的標簽信息和關鍵點坐標信息，其中，所述標簽信息為所述目標的類別信息；

文件生成模塊，用于根據所述彩色圖像對應的標簽信息和關鍵點坐標信息并基于對齊關系生成所述深度圖像對應的標注文件；

轉換模塊，用于將所述深度圖像轉換為偽彩色圖像；

標注模塊，用于基于所述深度圖像的偽彩色圖像對所述深度圖像對應的標注文件進行修正，得到修正后的標注數據集。

9.一種電子設備，包括存儲器、處理器以及存儲在所述存儲器中并可在所述處理器上運行的計算機程序，其特征在于，所述處理器執行所述計算機程序時實現如權利要求1至7任一項所述的圖像標注方法。

10.一種計算機存儲介質，所述計算機存儲介質存儲有計算機程序，其特征在于，所述計算機程序被處理器執行時實現如權利要求1至7任一項所述的圖像標注方法。