[發明專利]對象顏色的識別方法、裝置、計算機設備及存儲介質

有效

• 申請號: 202010108689.5
• 申請日: 2020-02-21
• 公開（公告）號: CN111340896B
• 公開（公告）日: 2023-10-27
• 發明（設計）人: 張保成
• 申請（專利權）人: 北京邁格威科技有限公司
• 主分類號: G06T7/90
• 分類號: G06T7/90;G06V10/764;G06V10/82
• 代理公司: 北京華進京聯知識產權代理有限公司 11606
• 代理人: 孫巖

本申請涉及一種對象顏色的識別方法、裝置、計算機設備及存儲介質。所述方法包括：獲取待識別對象的圖像；將所述待識別對象的圖像輸入至預先訓練的顏色分類模型中，輸出所述待識別對象的圖像包含的各局部圖像區域對應的候選顏色；針對每個局部圖像區域，如果該局部圖像區域對應的候選顏色的數目為多個，則根據該局部圖像區域對應的多個候選顏色和預設的顏色確定規則，確定該局部圖像區域對應的目標顏色；如果該局部圖像區域對應的候選顏色的數目為一種，則將該局部圖像區域對應的候選顏色，確定為該局部圖像區域對應的目標顏色。采用本申請可以提高對象顏色識別的準確性。

技術領域

本申請涉及計算機技術領域，特別是涉及一種對象顏色的識別方法、裝置、計算機設備及存儲介質。

背景技術

目前，在視頻結構化應用中，通常需要對視頻中目標對象的顏色進行識別。現有技術中，基于預設的顏色集合對樣本對象的圖像中包含的各局部圖像區域的顏色進行標注，并對神經網絡模型進行訓練。后續，通過訓練好的神經網絡模型識別對象的圖像中包含各局部圖像區域的顏色。

然而，預設的顏色集合中顏色數目有限，從而導致顏色識別不準確。

發明內容

基于此，有必要針對上述技術問題，提供一種對象顏色的識別方法、裝置、計算機設備及存儲介質。

第一方面，提供了一種對象顏色的識別方法，所述方法包括：

獲取待識別對象的圖像；

將所述待識別對象的圖像輸入至預先訓練的顏色分類模型中，輸出所述待識別對象的圖像包含的各局部圖像區域對應的候選顏色；

針對每個局部圖像區域，如果該局部圖像區域對應的候選顏色的數目為多個，則根據該局部圖像區域對應的多個候選顏色和預設的顏色確定規則，確定該局部圖像區域對應的目標顏色；

如果該局部圖像區域對應的候選顏色的數目為一種，則將該局部圖像區域對應的候選顏色，確定為該局部圖像區域對應的目標顏色。

作為一種可選地實施方式，所述將所述待識別對象的圖像輸入至預先訓練的顏色分類模型中，輸出所述待識別對象的圖像包含的各局部圖像區域對應的候選顏色，包括：

將所述待識別對象的圖像輸入至預先訓練的顏色分類模型中，輸出所述待識別對象的圖像包含的各局部圖像區域對應的各預設顏色的概率；

針對每個局部圖像區域，在所述各預設顏色中，將概率大于預設概率閾值的預設顏色，確定該局部圖像區域對應的候選顏色；或者，

針對每個局部圖像區域，在所述各預設顏色中，按照概率由大到小的順序，確定該局部圖像區域對應的預設數目個候選顏色。

作為一種可選地實施方式，所述各預設顏色包括無色調顏色或有色調顏色中的一種或多種。

作為一種可選地實施方式，該局部圖像區域對應的候選顏色的數目為兩個，所述根據該局部圖像區域對應的多個候選顏色和預設的顏色確定規則，確定該局部圖像區域對應的目標顏色，包括：

如果該局部圖像區域對應的兩個候選顏色為對比色或互補色，則將概率最大的候選顏色，確定為該局部圖像區域對應的目標顏色，所述候選顏色的概率是由所述預先訓練的顏色分類模型輸出的；

如果該局部圖像區域對應的兩個候選顏色為鄰近色或同類色，則將該局部圖像區域對應的兩個候選顏色的復合色，確定為該局部圖像區域對應的目標顏色。

作為一種可選地實施方式，當所述兩個候選顏色均為無色調顏色時，所述兩個候選顏色的復合色為灰度值在所述兩個候選顏色的灰度值之間的灰色；當所述兩個候選顏色均為有色調顏色時，所述兩個候選顏色的復合色為所述兩個候選顏色的混合顏色；當所述兩個候選顏色分別為無色調顏色和有色調顏色時，所述兩個候選顏色的復合色為所述兩種候選顏色的中間態顏色。