[發明專利]訓練樣本的生成方法、目標檢測方法和裝置

在審

• 申請號: 201811141521.3
• 申請日: 2018-09-28
• 公開（公告）號: CN109377467A
• 公開（公告）日: 2019-02-22
• 發明（設計）人: 錢佳景;王康;管維剛
• 申請（專利權）人: 阿里巴巴集團控股有限公司
• 主分類號: G06T5/50
• 分類號: G06T5/50;G06T7/13;G06K9/62
• 代理公司: 北京億騰知識產權代理事務所 11309
• 代理人: 孫欣欣;陳霽

1.一種訓練樣本的生成方法，所述方法包括：

獲取第一圖像，所述第一圖像中包括至少一個干擾物體；

獲取第二圖像，所述第二圖像中包括目標檢測物體，所述目標檢測物體具有四個頂點；

以所述第一圖像作為背景圖像以及以所述第二圖像作為前景圖像，將所述第一圖像和所述第二圖像進行合成處理，得到合成圖像；

獲取所述合成圖像中所述四個頂點的位置信息；

根據所述合成圖像和所述四個頂點的位置信息，生成神經網絡模型的一組訓練樣本，所述神經網絡模型用于從圖像中檢測目標物體的頂點。

2.如權利要求1所述的方法，其中，以所述第一圖像作為背景圖像以及以所述第二圖像作為前景圖像，將所述第一圖像和所述第二圖像進行合成處理，得到合成圖像，包括：

以所述第一圖像作為背景圖像，以所述第二圖像作為前景圖像，通過改變所述第二圖像的尺寸和/或所述第二圖像的位置，將所述第一圖像和所述第二圖像進行多次合成處理，得到多個合成圖像。

3.如權利要求2所述的方法，其中，所述獲取所述合成圖像中所述四個頂點的位置信息，包括：

獲取人工輸入的所述第二圖像中所述四個頂點的位置信息；

根據生成合成圖像時所述第二圖像的尺寸的變化，和/或所述第二圖像的位置變化，確定所述合成圖像中所述四個頂點的位置信息。

4.如權利要求2所述的方法，其中，所述獲取所述合成圖像中所述四個頂點的位置信息，包括：

獲取人工輸入的一個合成圖像中所述四個頂點的位置信息；

根據另一個合成圖像相對于該合成圖像中所述第二圖像的尺寸的變化，和/或另一個合成圖像相對于該合成圖像中所述第二圖像的位置變化，確定所述另一個合成圖像中所述四個頂點的位置信息。

5.如權利要求1所述的方法，其中，所述干擾物體包括具有干擾線段的物體。

6.如權利要求1所述的方法，其中，所述目標檢測物體為四邊形物體。

7.如權利要求1所述的方法，其中，所述根據所述合成圖像和所述四個頂點的位置信息，生成神經網絡模型的一組訓練樣本，包括：

通過邊緣檢測算法對所述合成圖像進行處理，生成包括所述目標檢測物體的邊緣信息的黑白圖像；

將所述黑白圖像和所述四個頂點的位置信息，作為神經網絡模型的一組訓練樣本。

8.一種目標檢測方法，所述方法包括：

獲取待檢測圖像，所述待檢測圖像中包含具有四個頂點的目標檢測物體；

將所述待檢測圖像輸入預先訓練的神經網絡模型，通過所述神經網絡模型的輸出得到所述目標檢測物體的四個頂點的位置信息，其中所述神經網絡模型利用權利要求1的方法生成的訓練樣本訓練得到。

9.如權利要求8所述的方法，其中，獲取待檢測圖像包括：

獲取原始圖像，所述原始圖像中包含具有四個頂點的目標檢測物體；

利用邊緣檢測算法對所述原始圖像進行處理，得到包含所述目標檢測物體的邊緣信息的黑白圖像作為所述待檢測圖像。

10.如權利要求8所述的方法，其中，所述神經網絡模型包含卷積神經網絡CNN。

11.一種訓練樣本的生成裝置，所述裝置包括：

獲取單元，用于獲取第一圖像，所述第一圖像中包括至少一個干擾物體；獲取第二圖像，所述第二圖像中包括目標檢測物體，所述目標檢測物體具有四個頂點；

合成單元，用于以所述獲取單元獲取的第一圖像作為背景圖像以及以所述獲取單元獲取的第二圖像作為前景圖像，將所述第一圖像和所述第二圖像進行合成處理，得到合成圖像；

所述獲取單元，還用于獲取所述合成單元得到的合成圖像中所述四個頂點的位置信息；

生成單元，用于根據所述合成單元得到的合成圖像和所述獲取單元獲取的四個頂點的位置信息，生成神經網絡模型的一組訓練樣本，所述神經網絡模型用于從圖像中檢測目標物體的頂點。