[發明專利]傾斜目標的處理方法、裝置、存儲介質與處理器

在審

• 申請號: 202210056957.2
• 申請日: 2022-01-18
• 公開（公告）號: CN114494398A
• 公開（公告）日: 2022-05-13
• 發明（設計）人: 肖傳利
• 申請（專利權）人: 深圳市聯洲國際技術有限公司
• 主分類號: G06T7/60
• 分類號: G06T7/60;G06T7/66;G06T7/70;G06N3/04;G06N3/08;G06N20/00
• 代理公司: 北京康信知識產權代理有限責任公司 11240
• 代理人: 霍文娟

本申請提供了一種傾斜目標的處理方法、裝置、存儲介質與處理器。該方法包括：獲取標注后的多個原始圖像，原始圖像中包括傾斜目標；確定各標注后的原始圖像對應的位置密度圖和方向密度圖；構建人工智能模型，人工智能模型為使用多組訓練數據通過機器學習訓練得到的，多組訓練數據中的每一組訓練數據均包括：原始圖像、原始圖像對應的位置密度圖和原始圖像對應的方向密度圖；將待預測的目標圖像輸入至人工智能模型中進行運算，輸出目標圖像對應的目標位置密度圖和目標方向密度圖；根據目標位置密度圖和目標方向密度圖，確定目標圖像中的傾斜目標的傾斜角度。本方案獲取了精確的傾斜目標的傾斜角度。

技術領域

本申請涉及圖像處理領域，具體而言，涉及一種傾斜目標的處理方法、裝置、存儲介質與處理器。

背景技術

在現在大量攝像頭使用的情況下，人臉檢測、行人檢測等功能被廣泛地應用在移動端的攝像頭上。在部分場景中，如艦船目標檢測，需要對傾斜目標進行檢測。如果使用傳統的水平矩形框表示方法，難以準確的獲取傾斜目標的角度信息。

發明內容

本申請的主要目的在于提供一種傾斜目標的處理方法、裝置、存儲介質與處理器，以解決現有技術中難以準確地獲取傾斜目標的角度信息的問題。

為了實現上述目的，根據本申請的一個方面，提供了一種傾斜目標的處理方法，包括：獲取標注后的多個原始圖像，所述原始圖像中包括傾斜目標；確定各所述標注后的原始圖像對應的位置密度圖和方向密度圖，所述位置密度圖在所述傾斜目標的中心點的取值，不同于在所述原始圖像中除所述中心點以外的點的取值，所述方向密度圖在方向標志點的取值不同于在所述原始圖像中除所述方向標志點以外的點的取值，所述方向標志點為所述傾斜目標上的非所述中心點的點；構建人工智能模型，所述人工智能模型為使用多組訓練數據通過機器學習訓練得到的，所述多組訓練數據中的每一組訓練數據均包括：所述原始圖像、所述原始圖像對應的位置密度圖和所述原始圖像對應的方向密度圖；將待預測的目標圖像輸入至所述人工智能模型中進行運算，輸出所述目標圖像對應的目標位置密度圖和目標方向密度圖；根據所述目標位置密度圖和目標方向密度圖，確定所述目標圖像中的傾斜目標的傾斜角度。

可選地，根據所述目標位置密度圖和目標方向密度圖，確定所述目標圖像中的傾斜目標的傾斜角度，包括：根據所述目標位置密度圖確定所述目標圖像中的傾斜目標的中心點的坐標；根據所述目標方向密度圖確定所述目標圖像中的傾斜目標的方向標志點的坐標；根據所述中心點的坐標和所述方向標志點的坐標確定所述目標圖像中的傾斜目標的傾斜角度。

可選地，根據所述目標位置密度圖確定所述目標圖像中的傾斜目標的中心點的坐標，包括：獲取所述目標位置密度圖的局部區域的第一極大值對應的坐標；確定所述第一極大值是否大于第一閾值；在所述第一極大值大于所述第一閾值的情況下，確定所述第一極大值對應的坐標為所述傾斜目標的中心點的坐標。

可選地，根據所述目標方向密度圖確定所述目標圖像中的傾斜目標的方向標志點的坐標，包括：獲取所述目標方向密度圖的第二極大值對應的坐標；確定所述第二極大值是否大于第二閾值；確定所述第二極大值對應的坐標是否在所述傾斜目標的中心點的坐標的鄰域范圍內；在所述第二極大值大于所述第二閾值，且所述第二極大值對應的坐標在所述鄰域范圍內的情況下，確定所述第二極大值對應的坐標為所述方向標志點的坐標。

可選地，所述訓練數據還包括所述原始圖像對應的尺寸密度圖，所述方法還包括：從標注后的所述原始圖像中，提取出所述原始圖像中的傾斜目標的中心點坐標；從標注后的所述原始圖像中，提取出所述原始圖像中的傾斜目標的尺寸；從標注后的所述原始圖像中，提取出所述原始圖像中的傾斜目標的傾斜角度；根據傾斜目標的中心點坐標、所述傾斜目標的尺寸和所述傾斜目標的傾斜角度，生成所述原始圖像對應的尺寸密度圖；將待預測的目標圖像輸入至所述人工智能模型中進行運算，輸出所述目標圖像對應的目標尺寸密度圖。

可選地，所述方法還包括：根據所述目標尺寸密度圖，確定所述目標圖像中的傾斜目標的長度信息和寬度信息。