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

在審

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

本申請涉及深度學習技術領域，提供了一種圖像標注方法、裝置及電子設備，該方法包括：獲取對齊的彩色圖像和深度圖像；利用預設的經訓練的目標檢測模型對彩色圖像進行檢測，獲得所述彩色圖像對應的標簽信息和關鍵點坐標信息，其中，所述標簽信息為所述目標的類別信息；根據所述彩色圖像對應的標簽信息和關鍵點坐標信息并基于對齊關系生成所述深度圖像對應的標注文件；將所述深度圖像轉換為偽彩色圖像，并基于所述深度圖像的偽彩色圖像對所述深度圖像對應的標注文件進行修正，得到修正后的標注數據集。本實施例提高了圖像標注效率。

技術領域

本申請涉及深度學習技術領域，尤其涉及一種圖像標注方法、裝置及電子設備。

背景技術

在深度學習中，需要利用大量的具有圖像標注信息的圖像作為數據集進行深度學習和訓練，但是數據標注是一件非常繁瑣和耗時的工作。

目前大多數研究機構或公司會采用購買現有的數據集，或者使用現有公開數據集的方式獲得數據集，但是這種方式不利于特定場景或特定目標檢測的研究。除此之外，他們也會采用外包給標注公司進行數據標注以獲得數據集的方式，這種方式使得進行深度學習研究的成本非常高。

目前，數據標注大多采用人工標注的方式，標注人員需要進行大量的重復性的判斷和操作來對圖像進行標注，這樣不可避免地需要投入大量的人力成本和時間成本，降低了數據集優化和模型建立的效率。

發明內容

有鑒于此，本申請實施例提供了一種圖像標注方法、裝置及電子設備，能夠解決相關技術中的一個或多個技術問題。

第一方面，本申請一實施例提供了一種圖像標注方法，包括：

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

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

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

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

本實施例，一方面，使用經訓練的公開模型對彩色圖像進行推理檢測，根據檢測結果生成標注文件，減少人工標注，大大提高了數據集的標注效率。另一方面，將深度圖像轉換為偽彩色圖像方便用戶對標注文件進行復檢，提高了復檢效率，也提高了標注的準確度，獲得了置信度更高的數據集。再一方面，將標注文件應用于深度圖像中，實現了快速標注深度圖像的標注方法，進而進行訓練學習，能夠快速開展對深度圖像的研究，促進3D相關技術的開發。

作為第一方面的一實現方式，所述將所述深度圖像轉換為偽彩色圖像，包括：

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

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

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

作為第一方面的一實現方式，所述獲取對齊的彩色圖像和深度圖像，包括：

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

作為第一方面的一實現方式，所述根據彩色圖像對應的標簽信息和關鍵點坐標信息并基于對齊關系生成深度圖像對應的標注文件，包括：

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

作為第一方面的一實現方式，所述標注文件還包括所述深度圖像中圖像信息，其中，所述圖像信息包括圖像的長、寬、通道、路徑和圖像名稱。