[發明專利]戶型形狀識別方法、裝置和設備

在審

• 申請號: 201910317211.0
• 申請日: 2019-04-17
• 公開（公告）號: CN110059750A
• 公開（公告）日: 2019-07-26
• 發明（設計）人: 王勝;施賢
• 申請（專利權）人: 廣東三維家信息科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62
• 代理公司: 北京超凡宏宇專利代理事務所(特殊普通合伙) 11463
• 代理人: 徐彥圣

本發明提供了一種戶型形狀識別方法、裝置和設備，涉及圖像處理的技術領域，該方法包括：獲取待識別的戶型圖；將戶型圖輸入至預先建立的戶型形狀識別模型中，輸出戶型圖的識別結果；其中，該戶型形狀識別模型通過下述方式建立：獲取設定數量的戶型圖樣本和對應的標識信息；建立卷積神經網絡的網絡結構；將戶型圖樣本、戶型形狀信息、以及標注在戶型圖樣本中的結構信息輸入至卷積神經網絡進行訓練；將訓練完成的神經網絡進行調優處理，得到訓練完成的戶型形狀識別模型，解決了現有技術中存在的當戶型圖數量較大時，采用現有人工方式識別效率和準確性低的技術問題，提高了戶型形狀識別的效率和準確性。

技術領域

本發明涉及圖像處理技術領域，尤其是涉及一種戶型形狀識別方法、裝置和設備。

背景技術

家裝設計是指家庭裝修設計，在家庭正式裝潢開工前進行功能格局上的規劃設計，各空間界面的裝飾設計。在家裝設計中，需要綜合考慮戶型圖房間形狀，戶型圖房間形狀主要包括：正方形、長方形、L形、多邊形、凹形、凸形等。

家裝設計前，首先需要得到戶型圖房間形狀圖像的幾何矢量數據，目前主要是通過人工識別手動生成這些數據，當戶型圖數量較大時，采用現有人工方式的識別效率和準確性低。

發明內容

本發明的目的在于提供一種戶型形狀識別方法及裝置，以提高戶型形狀識別的效率和準確性。

本發明提供的一種戶型形狀識別方法，所述方法包括：獲取待識別的戶型圖；將所述戶型圖輸入至預先建立的戶型形狀識別模型中，輸出所述戶型圖的識別結果；所述戶型形狀識別模型通過下述方式建立：獲取設定數量的戶型圖樣本和對應的標識信息；所述標識信息包括戶型形狀信息，以及標注在所述戶型圖樣本中的結構信息；建立卷積神經網絡的網絡結構；所述網絡結構包含有相互連接的第一子網絡和第二子網絡；將所述戶型圖樣本和標注在所述戶型圖樣本中的結構信息輸入至所述第一子網絡中進行訓練；將所述戶型圖樣本、戶型形狀信息、以及標注在所述戶型圖樣本中的結構信息輸入至所述第二子網絡中進行訓練；將訓練完成的所述第一子網絡和所述第二子網絡進行調優處理，得到訓練完成的戶型形狀識別模型。

進一步的，所述戶型形狀識別模型中的第一子網絡包括兩層卷積層，用于提取戶型圖的結構信息；所述結構信息包括點位置信息和邊位置信息；所述第二子網絡包括遞歸子網絡和卷積層，用于將所述結構信息編碼為張量，將所述張量映射為包含有類別信息的向量。

進一步的，將所述戶型圖輸入至預先建立的戶型形狀識別模型中的步驟之前，所述方法還包括：對所述戶型圖進行采樣處理，得到采樣后的所述戶型圖；對采樣后的所述戶型圖進行歸一化處理，得到歸一化處理后的所述戶型圖。

進一步的，得到訓練完成的戶型形狀識別模型的步驟之后，所述方法還包括：計算所述戶型形狀識別模型輸出的所述戶型圖樣本對應的識別結果；根據所述戶型圖樣本對應的標識信息和所述識別結果，計算所述戶型圖樣本的識別準確率；判斷所述識別準確率是否滿足預設閾值；如果否，繼續執行獲取設定數量的戶型圖樣本和對應的標識信息的步驟，直至所述識別準確率滿足預設閾值。

本發明提供的一種戶型形狀識別裝置，所述裝置包括：獲取模塊，用于獲取待識別的戶型圖；結果輸出模塊，用于將所述戶型圖輸入至預先建立的戶型形狀識別模型中，輸出所述戶型圖的識別結果；所述戶型形狀識別模型通過下述方式建立：獲取設定數量的戶型圖樣本和對應的標識信息；所述標識信息包括戶型形狀信息，以及標注在所述戶型圖樣本中的結構信息；建立卷積神經網絡的網絡結構；所述網絡結構包含有相互連接的第一子網絡和第二子網絡；將所述戶型圖樣本和標注在所述戶型圖樣本中的結構信息輸入至所述第一子網絡中進行訓練；將所述戶型圖樣本、戶型形狀信息、以及標注在所述戶型圖樣本中的結構信息輸入至所述第二子網絡中進行訓練；將訓練完成的所述第一子網絡和所述第二子網絡進行調優處理，得到訓練完成的戶型形狀識別模型。