[發明專利]影像數據檢測方法和裝置

有效

• 申請號: 202010367441.0
• 申請日: 2020-04-30
• 公開（公告）號: CN111583220B
• 公開（公告）日: 2023-04-18
• 發明（設計）人: 曹世磊;劉華羅;鄭冶楓
• 申請（專利權）人: 騰訊科技（深圳）有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06V10/764;G06V10/40;G06V10/82;G06N3/0464
• 代理公司: 華進聯合專利商標代理有限公司 44224
• 代理人: 王寧;李文淵

1.一種影像數據檢測方法，其特征在于，所述方法包括：

獲取待檢測影像，所述待檢測影像包括切片圖像序列；

對于切片圖像序列中的每個目標圖像，確定對應的切片圖像組，所述切片圖像組包括所述目標圖像、及在所述切片圖像序列中與所述目標圖像具有上下文關系的鄰近圖像；

對所述切片圖像組中的各切片圖像分別提取對應的切片特征圖；

獲取所述切片圖像組對應的切片數量，利用所述切片圖像組對應提取的各切片特征圖進行堆疊處理，堆疊后的特征圖包括切片數量維度和通道維度，將所述切片數量維度的特征與所述通道維度的特征進行交換，生成拼接后的特征圖，利用拼接后的特征圖進行對齊操作，生成對齊后的特征圖；

利用對齊后的特征圖對所述切片圖像組中的各切片圖像的上下文信息進行聚合；

對聚合后的特征圖進行目標區域檢測，得到與所述目標圖像對應的檢測結果，將每個目標圖像對應的檢測結果進行合并，生成與所述待檢測影像對應的檢測結果。

2.根據權利要求1所述的方法，其特征在于，所述拼接后的特征圖還包括切片尺寸維度；所述利用拼接后的特征圖進行對齊操作包括：

對所述拼接后的特征圖進行三維卷積操作；

對卷積操作后的特征圖在所述切片數量維度和所述切片尺寸維度上進行歸一化處理；

通過對歸一化后的特征圖進行非線性激活，輸出對齊后的特征圖。

3.根據權利要求1所述的方法，其特征在于，所述利用對齊后的特征圖對所述切片圖像組中的各切片圖像的上下文信息進行聚合包括：

根據切片數量維度將所述對齊后的特征圖進行重塑，生成重塑后的特征圖；

利用所述重塑后的特征圖對所述對齊后的特征圖進行加權；

對加權后的特征圖進行降維處理，生成與所述切片圖像組對應的聚合后的特征圖。

4.根據權利要求3所述的方法，其特征在于，所述利用所述重塑后的特征圖對所述對齊后的特征圖進行加權，對加權后的特征圖進行降維處理包括：

將所述重塑后的特征圖進行轉換，利用轉換后的特征圖對所述對齊后的特征圖的切片數量維度進行加權；

獲取降維卷積，利用所述降維卷積對所述加權后的特征圖進行降維處理。

5.根據權利要求4所述的方法，其特征在于，所述降維卷積的卷積核根據所述切片圖像組對應的切片數量生成。

6.根據權利要求1所述的方法，其特征在于，所述對聚合后的特征圖進行目標區域檢測包括：

識別所述聚合后的特征圖中落入目標檢測框內的像素點對應的類別；

回歸得到所述像素點的包圍框；

預測所述像素點距離目標檢測框中心點的相對距離，當所述相對距離超過預設距離時，過濾所述像素點的包圍框。

7.一種影像檢測模型訓練方法，其特征在于，所述影像檢測模型包括骨干網絡、三維對齊網絡、三維聚合網絡和目標檢測網絡，所述方法包括：

對于影像樣本的切片樣本圖像序列中的每個目標樣本圖像，確定對應的切片樣本圖像組；所述切片樣本圖像組包括所述目標樣本圖像、及在所述切片樣本圖像序列中與所述目標樣本圖像具有上下文關系的鄰近樣本圖像；

通過所述骨干網絡對所述切片樣本圖像組中的各切片樣本圖像分別提取對應的切片樣本特征圖；

通過所述三維對齊網絡將所述切片樣本圖像組對應提取的切片樣本特征圖進行對齊操作；

通過所述三維聚合網絡根據切片數量維度將所述對齊后的樣本特征圖進行重塑，生成重塑后的樣本特征圖；

利用所述重塑后的樣本特征圖對所述對齊后的樣本特征圖進行加權；

對加權后的樣本特征圖進行降維處理，生成與所述切片樣本圖像組對應的聚合后的樣本特征圖；

通過所述目標檢測網絡對聚合后的樣本特征圖進行目標區域檢測，得到與所述目標樣本圖像對應的檢測結果，將每個目標樣本圖像對應的檢測結果進行合并，生成與所述影像樣本對應的檢測結果。