[發明專利]一種圖像質量篩選方法及裝置、設備和存儲介質

在審

• 申請號: 201811116594.7
• 申請日: 2018-09-25
• 公開（公告）號: CN109345522A
• 公開（公告）日: 2019-02-15
• 發明（設計）人: 唐回峰;侯軍;伊帥
• 申請（專利權）人: 北京市商湯科技開發有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06K9/00
• 代理公司: 北京派特恩知識產權代理有限公司 11270
• 代理人: 蔣雅潔;張穎玲

本申請實施例提供一種圖像質量篩選方法及裝置、設備和存儲介質，其中，獲取待篩選圖像集；將所述待篩選圖像集輸入訓練好的篩選模型，得到圖像質量篩選結果；其中，所述篩選模型是通過根據樣本圖像所包含的人體關鍵點信息、樣本圖像清晰度進行訓練得到的；輸出所述圖像質量篩選結果。

技術領域

本申請實施例涉及計算機視覺通信領域，涉及但不限于一種圖像質量篩選方法及裝置、設備和存儲介質。

背景技術

行人檢索任務指給出一個行人的詢問圖片，需要在一個大規模的數據集中尋找到同一個人的所有圖片。解決這個問題主要是要對每一張行人圖片提取一個特征向量，利用這個特征向量將不同的行人區分開。

通常行人檢索的測試如下進行：給出probe和gallery兩個列表，記錄了不同的人的圖片以及對應的身份編號，其中probe為待檢索的行人庫，而gallery是被檢索的行人庫，通常稱為底庫。在測試時對probe和gallery中的每張圖片都通過行人檢索網絡提取出特征向量，將probe中的每個向量都在gallery中的向量構成的向量空間中進行搜索，將搜索結果按照兩個向量的相似度進行排序，排名越靠前意味著兩個向量越相近，即兩張圖片上的人是同一個人的概率越大.

但是在實際應用場景中，檢索會受到各種各樣因素的干擾：檢測框的完整度、行人之間的遮擋關系、行人的完整度、圖片的分辨率、光照等，在相關技術的模型中，如果同一個人的兩張圖片，具有不同程度的負面干擾，如第一張是該人全身的照片，第二張則是上半身的，那么在搜索時就會產生一些負面影響，導致結果不準確。

發明內容

有鑒于此，本申請實施例提供一種圖像質量篩選方法及裝置、設備和存儲介質。

本申請實施例的技術方案是這樣實現的：

本申請實施例提供一種圖像質量篩選方法，所述方法包括：

獲取待篩選圖像集；

將所述待篩選圖像集輸入訓練好的篩選模型，得到圖像質量篩選結果；其中，所述篩選模型是通過根據樣本圖像所包含的人體關鍵點信息、樣本圖像清晰度進行訓練得到的；

輸出所述圖像質量篩選結果。

在本申請實施例中，所述篩選模型的訓練過程，包括：

根據獲取的樣本圖像集的關鍵點信息和所述樣本圖像集的清晰度，確定所述樣本圖像集中滿足預設條件的第一樣本圖像集；

確定所述第一樣本圖像集中與預設的待搜索圖像集所包含的圖像相關的第二樣本圖像集；

根據所述第二樣本圖像集完成對所述神經網絡模型的訓練。

在本申請實施例中，所述確定所述第一樣本圖像集中與預設的待搜索圖像集所包含的圖像相關的第二樣本圖像集，包括：

在所述第一樣本圖像集中搜索所述待搜索圖像集的每一圖像，得到在所述第一樣本圖像集中的所述每一圖像對應的順序列表；

根據所述順序列表，確定所述待搜索圖像集的每一圖像對應的平均正確率，得到平均正確率集；

確定所述平均正確率集中大于預設閾值的平均正確率對應的第二樣本圖像集。

在本申請實施例中，所述樣本圖像集中的圖像包括至少包含一個行人的圖像或者視頻；所述關鍵點信息包括所述樣本圖像中行人的人體部位。

在本申請實施例中，所述根據獲取的樣本圖像集的關鍵點信息和所述樣本圖像集的清晰度，確定所述樣本圖像集中滿足預設條件的第一樣本圖像集，包括：

根據所述樣本圖像集中每一圖像包含的行人的關鍵點信息，確定所述每一圖像中行人的人體完整度信息；