[發明專利]一種遮擋條件下的工裝顏色識別方法及系統

在審

• 申請號: 202110275035.6
• 申請日: 2021-03-15
• 公開（公告）號: CN112990012A
• 公開（公告）日: 2021-06-18
• 發明（設計）人: 柯海濱
• 申請（專利權）人: 深圳喜為智慧科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/34;G06K9/46;G06K9/62
• 代理公司: 暫無信息
• 代理人: 暫無信息

1.一種遮擋條件下的工裝顏色識別方法，其特征在于，包括：

響應于識別到圖像中的行人被遮擋，根據所述行人在多個視頻幀中的分割圖得到無遮擋的完整圖和遮擋的存疑區域；

將所述完整圖和存疑區域輸入預先訓練的深度學習顏色分類模型，得到所述行人穿著工裝的顏色識別結果。

2.根據權利要求1所述的方法，其特征在于，識別圖像中的行人是否被遮擋，具體包括：

利用有遮擋和無遮擋的數據集訓練有無遮擋分類模型；

將圖像輸入所述有無遮擋分類模型，確定所述圖像中的行人是否被遮擋。

3.根據權利要求1所述的方法，其特征在于，根據所述行人在多個視頻幀中的分割圖得到無遮擋的完整圖，具體包括：

通過圖像補全技術對所述行人在多個視頻幀中的分割圖進行補全，得到無遮擋的完整圖。

4.根據權利要求1所述的方法，其特征在于，根據所述行人在多個視頻幀中的分割圖得到遮擋的存疑區域，具體包括：

將所述分割圖與完整圖相減，得到遮擋的存疑區域。

5.根據權利要求1-4任一項所述的方法，其特征在于，所述分割圖的獲取過程，具體包括：

獲取預設時間間隔的多個視頻幀；

利用多目標跟蹤技術，對所述多個視頻幀中的每個行人分配身份標簽；

將所述多個視頻幀輸入實例分割模型，獲取所述多個視頻幀中同一身份標簽的行人的分割圖。

6.一種遮擋條件下的工裝顏色識別系統，其特征在于，包括：

圖像處理模塊，用于響應于識別到圖像中的行人被遮擋，根據所述行人在多個視頻幀中的分割圖得到無遮擋的完整圖和遮擋的存疑區域；

顏色識別模塊，用于將所述完整圖和存疑區域輸入預先訓練的深度學習顏色分類模型，得到所述行人穿著工裝的顏色識別結果。

7.根據權利要求6所述的系統，其特征在于，還包括遮擋識別模塊，用于識別圖像中的行人是否被遮擋，具體包括：

模型訓練單元，用于利用有遮擋和無遮擋的數據集訓練有無遮擋分類模型；

遮擋識別單元，用于將圖像輸入所述有無遮擋分類模型，確定所述圖像中的行人是否被遮擋。

8.根據權利要求6所述的系統，其特征在于，所述圖像處理模塊，具體包括：

圖像補全單元，用于通過圖像補全技術對所述行人在多個視頻幀中的分割圖進行補全，得到無遮擋的完整圖。

9.根據權利要求6所述的系統，其特征在于，所述圖像處理模塊，具體還包括：

遮擋區域獲取單元，用于將所述分割圖與完整圖相減，得到遮擋的存疑區域。

10.根據權利要求6-9任一項所述的系統，其特征在于，還包括分割圖處理模塊，用于獲取分割圖，具體包括：

視頻幀獲取單元，用于獲取預設時間間隔的多個視頻幀；

目標跟蹤單元，用于利用多目標跟蹤技術，對所述多個視頻幀中的每個行人分配身份標簽；

實例分割單元，用于將所述多個視頻幀輸入實例分割模型，獲取所述多個視頻幀中同一身份標簽的行人的分割圖。