[發(fā)明專利]一種基于目標檢測學習算法的圖片數(shù)據(jù)篩選方法及系統(tǒng)

在審

• 申請?zhí)枺?/td>: 202210568801.2
• 申請日: 2022-05-24
• 公開（公告）號: CN114898182A
• 公開（公告）日: 2022-08-12
• 發(fā)明（設計）人: 閆軍;紀雙西
• 申請（專利權）人: 智慧互通科技股份有限公司
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V10/25;G06V10/42;G06V10/762;G06F16/538
• 代理公司: 暫無信息
• 代理人: 暫無信息

1.一種基于目標檢測學習算法的圖片數(shù)據(jù)篩選方法，其特征在于，所述方法包括：

將初始待標注圖片數(shù)據(jù)集劃分為訓練集、驗證集、測試集；

通過所述訓練集和所述驗證集對初始目標檢測模型進行訓練，獲取目標檢測模型；

根據(jù)所述測試集、所述目標檢測模型的綜合評分指標、所述目標檢測模型的預測結果，確定所述初始待標注圖片數(shù)據(jù)集對應的綜合圖片不確定性指標，并根據(jù)所述綜合圖片不確定性指標對所述初始待標注圖片數(shù)據(jù)集進行篩選；

通過所述目標檢測模型和預置圖片特征聚類算法對篩選后的待標注圖片數(shù)據(jù)進行二次篩選；

將所述進行二次篩選后的待標注圖片數(shù)據(jù)集通過所述目標檢測模型進行預測，獲取預標注標簽；

根據(jù)所述預標注標簽對待標注圖片數(shù)據(jù)集進行修正，并通過修正后的待標注圖片數(shù)據(jù)集更新所述訓練集后，訓練所述目標檢測模型。

2.根據(jù)權利要求1所述的一種基于目標檢測學習算法的圖片數(shù)據(jù)篩選方法，其特征在于，所述根據(jù)所述測試集、所述目標檢測模型的綜合評分指標、所述目標檢測模型的預測結果，確定所述初始待標注圖片數(shù)據(jù)集對應的綜合圖片不確定性指標的步驟包括：

通過所述綜合評分指標的上下限閾值獲取兩組預測結果，并根據(jù)兩組預測結果獲取圖片級目標框定位不穩(wěn)定性指標；

通過所述綜合評分指標和目標檢測模型在預期最終輸出條件下的評分閾值獲取的預測結果，獲取圖片級分類預測不穩(wěn)定性指標；

根據(jù)所述測試集的標注信息，獲取所述預測結果相對于標注信息的圖片級平均準確率；

通過所述圖片級目標框定位不穩(wěn)定性指標、圖片級分類預測不穩(wěn)定性指標以及圖片級平均準確率，確定綜合圖片不確定性指標。

3.根據(jù)權利要求2所述的一種基于目標檢測學習算法的圖片數(shù)據(jù)篩選方法，其特征在于，所述通過所述圖片級目標框定位不穩(wěn)定性指標、圖片級分類預測不穩(wěn)定性指標以及圖片級平均準確率，確定綜合圖片不確定性指標，并根據(jù)所述綜合圖片不確定性指標對所述初始待標注圖片數(shù)據(jù)集進行篩選的步驟包括：

根據(jù)所有測試集圖片分別對應的圖片級目標框定位不穩(wěn)定性指標、圖片級分類預測不穩(wěn)定性指標以及圖片級平均準確率，獲取各個圖片分別對應的綜合圖片不確定性指標；

獲取綜合圖片不確定性指標大于預設閾值的圖片作為篩選后的初始待標注圖片數(shù)據(jù)集。

4.根據(jù)權利要求1或2所述的一種基于目標檢測學習算法的圖片數(shù)據(jù)篩選方法，其特征在于，所述通過所述目標檢測模型和預置圖片特征聚類算法對篩選后的待標注數(shù)據(jù)進行二次篩選的步驟包括：

將所述待標注圖片數(shù)據(jù)輸入到所述目標檢測模型，提取模型FPN層輸出的多尺度特征圖，并將所述多尺度特征圖進行池化拼接后得到每個待標注圖片對應的特征向量；

將所述所有圖片的特征向量組合的特征向量矩陣進行壓縮得到預期維度的特征表示，并通過對所有圖片的特征表示集合進行聚類計算，獲取各個圖片的聚類簇類別；

根據(jù)各個圖片的聚類簇類別對所述待標注圖片數(shù)據(jù)集進行二次篩選。

5.根據(jù)權利要求4所述的一種基于目標檢測學習算法的圖片數(shù)據(jù)篩選方法，其特征在于，所述根據(jù)各個圖片的聚類簇類別對所述待標注圖片數(shù)據(jù)集進行二次篩選的步驟包括：

根據(jù)各個圖片的聚類簇類別標簽,計算經(jīng)過第一次篩選后的數(shù)據(jù)集中各個聚類簇內(nèi)數(shù)據(jù)量占總數(shù)據(jù)量的比值以及各個簇內(nèi)二次篩選數(shù)據(jù)量；

根據(jù)各個聚類簇內(nèi)數(shù)據(jù)分別對應的綜合圖片不確定性指標，從大到小選取預置個數(shù)的數(shù)據(jù)進行整合作為最終的二次篩選數(shù)據(jù)集。