[發明專利]適用于工業部件識別模型的訓練方法、裝置及存儲介質

有效

• 申請號: 202110197009.6
• 申請日: 2021-02-22
• 公開（公告）號: CN112926432B
• 公開（公告）日: 2023-08-15
• 發明（設計）人: 林振峰
• 申請（專利權）人: 杭州優工品科技有限公司
• 主分類號: G06V20/64
• 分類號: G06V20/64;G06V10/774;G06Q50/04;G06T3/00
• 代理公司: 北京知果之信知識產權代理有限公司 11541
• 代理人: 卜榮麗;李志剛

1.一種適用于工業部件識別模型的訓練方法,其特征在于，包括：

獲取工業部件數據庫中的預設工業部件三維圖像；

對所述預設工業部件三維圖像進行二維處理生成多個預設工業部件二維圖像，其中每個預設工業部件二維圖像為同一工業部件的不同角度的圖像；

將所有同一工業部件的不同角度的預設工業部件二維圖像作為訓練樣本對工業部件識別模型進行訓練；

獲取用戶當前時刻定制的定制工業部件三維圖像；

對所述定制工業部件三維圖像進行二維處理生成多個定制工業部件二維圖像，其中每個定制工業部件二維圖像為同一工業部件的不同角度的圖像；

將所有同一工業部件的不同角度的定制工業部件二維圖像作為訓練樣本對工業部件識別模型再次進行訓練；

接收使用者輸入的定制工業部件信息；

將所述定制工業部件信息與所述定制工業部件三維圖像相對應設置，當工業部件識別模型識別到二維圖像與所述定制工業部件三維圖像相對應時，輸出所述定制工業部件三維圖像的同時輸出所述定制工業部件信息。

2.根據權利要求1所述的訓練方法,其特征在于，

所述獲取用戶當前時刻定制的定制工業部件三維圖像包括：

判斷所述定制工業部件三維圖像與數據庫中的預設工業部件三維圖像是否相同；

若不存在與所述定制工業部件三維圖像相同的預設工業部件三維圖像，則對所述定制工業部件三維圖像進行二維處理生成多個定制工業部件二維圖像。

3.根據權利要求1至2中任意一項所述的訓練方法,其特征在于，

其中，所述預設工業部件三維圖像具有與其對應的預設工業部件信息；

當工業部件識別模型識別到二維圖像與所述預設工業部件三維圖像相對應時，輸出所述預設工業部件三維圖像的同時輸出所述預設工業部件信息。

4.一種適用于工業部件識別模型的訓練裝置，其特征在于，包括：

預設圖像獲取模塊，用于獲取工業部件數據庫中的預設工業部件三維圖像；

預設圖像處理模塊，用于對所述預設工業部件三維圖像進行二維處理生成多個預設工業部件二維圖像，其中每個預設工業部件二維圖像為同一工業部件的不同角度的圖像；

預設訓練模塊，用于將所有同一工業部件的不同角度的預設工業部件二維圖像作為訓練樣本對工業部件識別模型進行訓練；

定制圖像獲取模塊，用于獲取用戶當前時刻定制的定制工業部件三維圖像；

定制圖像處理模塊，用于對所述定制工業部件三維圖像進行二維處理生成多個定制工業部件二維圖像，其中每個定制工業部件二維圖像為同一工業部件的不同角度的圖像；

定制訓練模塊，用于將所有同一工業部件的不同角度的定制工業部件二維圖像作為訓練樣本對工業部件識別模型再次進行訓練；

接收單元，用于接收使用者輸入的定制工業部件信息；

輸出單元，用于將所述定制工業部件信息與所述定制工業部件三維圖像相對應設置，當工業部件識別模型識別到二維圖像與所述定制工業部件三維圖像相對應時，輸出所述定制工業部件三維圖像的同時輸出所述定制工業部件信息。

5.根據權利要求4所述的訓練裝置,其特征在于，

所述定制圖像獲取模塊包括：

判斷單元，用于判斷所述定制工業部件三維圖像與數據庫中的預設工業部件三維圖像是否相同；

處理單元，用于若不存在與所述定制工業部件三維圖像相同的預設工業部件三維圖像，則對所述定制工業部件三維圖像進行二維處理生成多個定制工業部件二維圖像。

6.一種可讀存儲介質，其特征在于，所述可讀存儲介質中存儲有計算機程序，所述計算機程序被處理器執行時用于實現權利要求1至3任一所述的方法。