[發明專利]小樣本涂裝分類器訓練方法、對象檢測方法以及電子設備

在審

• 申請號: 202110697613.5
• 申請日: 2021-06-23
• 公開（公告）號: CN113435298A
• 公開（公告）日: 2021-09-24
• 發明（設計）人: 姚達琛;劉皓松;李誠
• 申請（專利權）人: 上海商湯臨港智能科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62
• 代理公司: 深圳市威世博知識產權代理事務所(普通合伙) 44280
• 代理人: 何倚雯

1.一種小樣本涂裝分類器訓練方法，其特征在于，所述小樣本涂裝分類器訓練方法包括：

獲取涂裝對象的第一訓練圖像；

將所述第一訓練圖像輸入預先訓練的大樣本涂裝分類器，獲取所述涂裝對象的第一訓練特征；

利用所述第一訓練特征訓練所述小樣本涂裝分類器；

其中，所述大樣本涂裝分類器用于提取圖像中所述涂裝對象的特征，所述小樣本涂裝分類器用于利用所述涂裝對象的特征對所述涂裝對象進行分類。

2.根據權利要求1所述的小樣本涂裝分類器訓練方法，其特征在于，

所述將所述第一訓練圖像輸入預先訓練的大樣本涂裝分類器，獲取所述涂裝對象的第一訓練特征，包括：

將所述第一訓練圖像輸入預先訓練的對象外形檢測器，獲取所述涂裝對象的第一對象框圖像；

將所述第一對象框圖像輸入所述大樣本涂裝分類器，獲取所述第一對象框圖像的第一訓練特征。

3.根據權利要求2所述的小樣本涂裝分類器訓練方法，其特征在于，

所述將所述第一對象框圖像輸入所述大樣本涂裝分類器之前，包括：

將所述第一對象框圖像輸入預先訓練的關鍵點檢測器，獲取所述第一對象框圖像中的第一關鍵點和第二關鍵點；

旋轉所述第一對象框圖像，直至第二對象框圖像中第一關鍵點至第二關鍵點的射線方向朝向預設方向，其中，所述第二對象框圖像為旋轉后的所述第一對象框圖像。

4.根據權利要求1所述的小樣本涂裝分類器訓練方法，其特征在于，

所述將所述第一訓練圖像輸入預先訓練的大樣本涂裝分類器之前，包括：

將所述第一訓練圖像輸入預先訓練的關鍵點檢測器，獲取所述第一訓練圖像中的第一關鍵點和第二關鍵點；

旋轉所述第一訓練圖像，直至第二訓練圖像中第一關鍵點至第二關鍵點的射線方向朝向預設方向，其中，所述第二訓練圖像為旋轉后的所述第一訓練圖像。

5.根據權利要求3或4所述的小樣本涂裝分類器訓練方法，其特征在于，

所述第一關鍵點為車頭中心點，所述第二關鍵點為車尾中心點。

6.根據權利要求1-5任一項所述的小樣本涂裝分類器訓練方法，其特征在于，

所述獲取沙盤上第一涂裝對象的第一訓練圖像，包括：

對設置在四角位置和/或中心位置的多個所述涂裝對象進行采集，獲取多個所述涂裝對象的第一訓練圖像。

7.一種對象檢測方法，其特征在于，所述對象檢測方法包括：

獲取涂裝對象的第一檢測圖像；

將所述第一檢測圖像輸入預先訓練的大樣本涂裝分類器，獲取所述涂裝對象的第一檢測特征；

將所述第一檢測特征輸入小樣本涂裝分類器，獲取所述小樣本涂裝分類器對所述涂裝對象的分類結果；

其中，所述小樣本涂裝分類器通過權利要求1～6中任一項所述的小樣本涂裝分類器訓練方法訓練得到。

8.根據權利要求7所述的對象檢測方法，其特征在于，

所述將所述第一檢測圖像輸入預先訓練的大樣本涂裝分類器，獲取所述涂裝對象的第一檢測特征，包括：

將所述第一檢測圖像輸入預先訓練的對象外形檢測器，獲取所述涂裝對象的第一對象框圖像；

將所述第一對象框圖像輸入所述大樣本涂裝分類器，獲取所述第一對象框圖像的第一檢測特征。

9.根據權利要求8所述的對象檢測方法，其特征在于，

所述將所述第一對象框圖像輸入所述大樣本涂裝分類器之前，包括：

將所述第一對象框圖像輸入預先訓練的關鍵點檢測器，獲取所述第一對象框圖像中的第一關鍵點和第二關鍵點；

旋轉所述第一對象框圖像，直至第二對象框圖像中第一關鍵點至第二關鍵點的射線方向朝向預設方向，其中，所述第二對象框圖像為旋轉后的所述第一對象框圖像。