[發明專利]模型訓練方法、圖片方向糾正方法及裝置

在審

• 申請號: 202010114088.5
• 申請日: 2020-02-24
• 公開（公告）號: CN111563407A
• 公開（公告）日: 2020-08-21
• 發明（設計）人: 趙高攀
• 申請（專利權）人: 北京樂學幫網絡技術有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/32;G06N3/04;G06N3/08
• 代理公司: 北京超凡宏宇專利代理事務所(特殊普通合伙) 11463
• 代理人: 李飛

本申請實施例提供一種模型訓練方法、圖片方向糾正方法及裝置，該圖片方向糾正方法包括：獲取原圖片，其中，所述原圖片包括參照對象和待旋轉對象；通過參照對象識別網絡從所述原圖片上識別所述參照對象所在的區域；通過旋轉方向識別網絡識別所述區域內的參照對象的旋轉方向，其中，所述旋轉方向為所述原圖片的旋轉方向。本申請實施例能夠有針對性的、高準確率的識別拍攝方向不合適的圖片的方向，提高需要圖片旋轉場景下的圖片旋轉的效率。

技術領域

本申請涉及圖片方向旋轉領域，特別涉及一種模型訓練方法、圖片方向糾正方法及裝置。

背景技術

隨著網絡技術的發展，人們需要利用終端設備來拍攝圖片并通過網絡將圖片互相發送。由于終端拍攝角度的差異，導致圖片接收方接收的圖片角度不適合，進而影響圖片接受者的下一步工作。例如，目前在線教育領域，學生會以圖片方式向老師提交自己完成的作業，老師會對學生提交的圖片進行批改和標注，然而學生手寫練習的圖片角度各異(學生提交的圖片多為移動終端所拍攝，其拍攝的圖片角度會隨移動終端的角度而定)，這造成了為了完成作業批注，老師不可避免的需要完成作業圖片的角度糾正。具體地，老師通過肉眼觀察接收的作業圖片的角度并確定需要旋轉的角度，之后再由老師來完成圖片旋轉。

發明內容

本發明實施例提供一種模型訓練方法、圖片方向糾正方法及裝置，通過本申請實施例的技術方案能夠自動糾正拍攝的圖片方向，提高圖片接收方(例如，老師)對圖片進一步處理的工作效率。

第一方面，本申請實施例提供一種模型訓練方法，所述訓練方法包括根據多張包含參照對象和待旋轉對象的圖像訓練參照對象識別網絡，其中，所述參照對象用于攜帶所述待旋轉對象的方向信息；根據多張參照對象圖片訓練旋轉方向識別網絡。

通過本申請實施例的上述方法可以訓練得到兩個網絡，其中一個網絡用于參照對象的識別的目標識別網絡，另一個是基于目標識別網絡識別出的參照對象進行旋轉方式識別的旋轉方向識別網絡。本申請實施例的通過訓練好的兩個網絡協同工作完成了圖片旋轉方向識別，由于訓練網絡時采用的是反應待旋轉對象的參照對象作為特征數據因此有效減少了旋轉方向識別的數據處理量，節省了處理設備的存儲容量且提升了處理速度。

結合第一方面，在可能的實現方法中，所述參照對象識別網絡包括目標識別神經網絡；所述旋轉方向識別網絡為卷積神經網絡。

本申請實施例通過神經網絡對參照對象進行目標識別，對識別出的參照對象進行旋轉方向識別，可以針對不同場景來獲取不同的數據集來訓練神經網絡，提升了本申請實施例技術方案的應用場景。

結合第一方面，在可能的實現方法中，所述模型訓練方法還包括：統計所述旋轉方向識別網絡識別錯誤的圖片；根據所述圖片對所述旋轉方向識別網絡進行增強訓練。

本申請實施例通過統計旋轉方向識別網絡識別錯誤的圖片來進一步訓練該旋轉方向識別網絡，進一步提升了旋轉方向識別網絡對圖片旋轉方向識別的準確率。

第二方面，本申請實施例提供一種圖片方向糾正方法，所述方法包括：獲取原圖片，其中，所述原圖片包括參照對象和待旋轉對象；通過參照對象識別網絡從所述原圖片上識別所述參照對象所在的區域；通過旋轉方向識別網絡識別所述區域內的參照對象的旋轉方向，其中，所述旋轉方向為所述原圖片的旋轉方向。

本申請實施例通過識別參照對象的旋轉方向來確定原圖片的旋轉方向，由于參照對象數據處理量較少，因此可以提升旋轉方向識別網絡對旋轉方向識別的速度。

結合第二方面，在可能的實現方法中，所述參考對象包括二維碼或條形碼；所述通過旋轉方向識別網絡識別所述區域內的參照對象的旋轉方向，包括：所述旋轉方向識別網絡通過識別所述區域內的二維碼或者條形碼的點陣數據特征識別所述參照對象的旋轉方向。