本技術(shù)涉及圖像處理，特別是涉及一種基于深度學(xué)習(xí)的變焦方法、設(shè)備以及存儲介質(zhì)。

背景技術(shù)：

1、隨著圖像處理領(lǐng)域以及計算機視覺領(lǐng)域的相關(guān)技術(shù)的飛速發(fā)展，對圖像的采集與傳輸已經(jīng)普及到生活的各個方面。

2、尤其在某些特殊場景下，比如鐵路道路檢測、電力輸電線路檢測或者管道檢測、儀器儀表讀數(shù)檢測等諸多工業(yè)場景下，需要在遠距離、大倍率下仍舊能夠快速清晰地聚焦于采集圖像的感興趣區(qū)域中的目標對象。

3、但是受限于鏡頭的光學(xué)成像原理，尤其是長焦鏡頭往往使得采集圖像中的目標對象和背景無法同時保持聚焦清晰，通?？杀憩F(xiàn)為聚焦評價函數(shù)存在雙波峰的情況。在用戶使用過程中，則可能導(dǎo)致用戶原本期望聚焦于采集圖像中的前景目標對象，但實際鏡頭聚焦到圖像遠景的問題，降低變焦準確性。

技術(shù)實現(xiàn)思路

1、本技術(shù)至少提供一種基于深度學(xué)習(xí)的變焦方法、裝置、設(shè)備以及計算機可讀存儲介質(zhì)。

2、本技術(shù)第一方面提供了一種基于深度學(xué)習(xí)的變焦方法，包括：將圖像采集設(shè)備采集到的目標圖像輸入預(yù)先訓(xùn)練的擴散模型中，得到所述擴散模型輸出的聚焦預(yù)測圖像；對比所述目標圖像中的目標聚焦區(qū)域與所述聚焦預(yù)測圖像中的預(yù)測聚焦區(qū)域，得到聚焦對比結(jié)果；根據(jù)所述聚焦對比結(jié)果對所述圖像采集設(shè)備進行變焦處理。

3、在一實施例中，在所述將圖像采集設(shè)備采集到的目標圖像輸入預(yù)先訓(xùn)練的擴散模型中，得到所述擴散模型輸出的聚焦預(yù)測圖像的步驟之前，所述方法還包括：分別對圖像采集設(shè)備采集到的第一圖像和第二圖像進行聚焦評價處理，得到所述第一圖像對應(yīng)的第一聚焦評價值和所述第二圖像對應(yīng)的第二聚焦評價值；其中，所述第一圖像和所述第二圖像的圖像采集焦距不同；根據(jù)所述第一聚焦評價值、所述第二聚焦評價值和預(yù)設(shè)聚焦評價閾值判斷所述第一聚焦評價值和所述第二聚焦評價值之間是否存在聚焦評價峰值；響應(yīng)于所述第一聚焦評價值和所述第二聚焦評價值之間存在所述聚焦評價峰值，將所述第一圖像或所述第二圖像確定為所述目標圖像。

4、在一實施例中，在所述對比所述目標圖像中的目標聚焦區(qū)域與所述聚焦預(yù)測圖像中的預(yù)測聚焦區(qū)域的步驟之前，所述方法還包括：根據(jù)所述目標圖像的清晰度確定所述目標圖像中的目標聚焦區(qū)域，以及根據(jù)所述聚焦預(yù)測圖像的清晰度確定所述聚焦預(yù)測圖像中的預(yù)測聚焦區(qū)域；所述對比所述目標圖像中的目標聚焦區(qū)域與所述聚焦預(yù)測圖像中的預(yù)測聚焦區(qū)域，得到聚焦對比結(jié)果的步驟，包括：對比所述目標聚焦區(qū)域的位置信息和所述預(yù)測聚焦區(qū)域的位置信息，得到所述目標聚焦區(qū)域和所述預(yù)測聚焦區(qū)域之間的位置偏差，并根據(jù)所述位置偏差確定所述聚焦對比結(jié)果。

5、在一實施例中，所述根據(jù)所述目標圖像的清晰度確定所述目標圖像中的目標聚焦區(qū)域，以及根據(jù)所述聚焦預(yù)測圖像的清晰度確定所述聚焦預(yù)測圖像中的預(yù)測聚焦區(qū)域的步驟，包括：分別對所述目標圖像和所述聚焦預(yù)測圖像進行圖像分割處理，得到所述目標圖像中的多個目標子圖像以及所述聚焦預(yù)測圖像中的多個聚焦預(yù)測子圖像；將各個目標子圖像中的清晰度最高的目標子圖像確定為所述目標聚焦區(qū)域，以及將各個聚焦預(yù)測子圖像中的清晰度最高的聚焦預(yù)測子圖像確定所述預(yù)測聚焦區(qū)域。

6、在一實施例中，所述根據(jù)所述位置偏差確定所述聚焦對比結(jié)果的步驟，包括：響應(yīng)于所述位置偏差大于預(yù)設(shè)偏差閾值，判定所述圖像采集設(shè)備聚焦失敗，得到所述聚焦對比結(jié)果；響應(yīng)于所述位置偏差小于或等于所述預(yù)設(shè)偏差閾值，判定所述圖像采集設(shè)備聚焦成功，得到所述聚焦對比結(jié)果。

7、在一實施例中，所述根據(jù)所述聚焦對比結(jié)果對所述圖像采集設(shè)備進行變焦處理的步驟，包括：響應(yīng)于所述聚焦對比結(jié)果表征所述圖像采集設(shè)備聚焦失敗，根據(jù)預(yù)設(shè)變焦步長對所述圖像采集設(shè)備進行變焦處理，并獲取所述圖像采集設(shè)備在變焦處理之后獲取到第三圖像和第四圖像；其中，所述第三圖像和所述第四圖像的圖像采集焦距不同；分別對所述第三圖像和第四圖像進行聚焦評價處理，得到所述第三圖像對應(yīng)的第三聚焦評價值和所述第四圖像對應(yīng)的第四聚焦評價值；響應(yīng)于所述第三聚焦評價值和所述第四聚焦評價值之間存在峰值，縮小所述預(yù)設(shè)變焦步長，得到縮小后的變焦步長；根據(jù)所述縮小后的變焦步長對所述圖像采集設(shè)備進行變焦處理，直至所述圖像采集設(shè)備采集到的第五圖像和第六圖像之間的聚焦評價差值小于預(yù)設(shè)的聚焦誤差閾值。

8、在一實施例中，所述根據(jù)所述縮小后的變焦步長對所述圖像采集設(shè)備進行變焦處理，直至所述圖像采集設(shè)備采集到的第五圖像和第六圖像之間的聚焦評價差值小于預(yù)設(shè)的聚焦誤差閾值的步驟，包括：根據(jù)所述縮小后的變焦步長調(diào)整所述圖像采集設(shè)備的圖像采集焦距，得到調(diào)整后的圖像采集焦距；控制所述圖像采集設(shè)備根據(jù)所述調(diào)整后的圖像采集焦距進行圖像采集，得到所述第五圖像和所述第六圖像；其中，所述第五圖像和所述第六圖像的圖像采集焦距不同。

9、在一實施例中，在所述控制所述圖像采集設(shè)備根據(jù)所述調(diào)整后的圖像采集焦距進行圖像采集，得到所述第五圖像和所述第六圖像的步驟之后，所述方法還包括：分別對所述第五圖像和所述第六圖像進行聚焦評價處理，得到所述第五圖像對應(yīng)的第五聚焦評價值和所述第六圖像對應(yīng)的第六聚焦評價值；根據(jù)所述第五聚焦評價值和所述第六聚焦評價值確定所述聚焦評價差值；響應(yīng)于所述聚焦評價差值小于或等于所述聚焦誤差閾值，暫停或停止對所述圖像采集設(shè)備進行變焦處理

10、在一實施例中，所述分別對圖像采集設(shè)備采集到的第一圖像和第二圖像進行聚焦評價處理，得到所述第一圖像對應(yīng)的第一聚焦評價值和所述第二圖像對應(yīng)的第二聚焦評價值的步驟，包括：獲取所述第一圖像的灰度值；根據(jù)預(yù)設(shè)的聚焦評價函數(shù)和所述第一圖像的灰度值進行計算，得到所述第一聚焦評價值。

11、在一實施例中，在所述將圖像采集設(shè)備采集到的目標圖像輸入預(yù)先訓(xùn)練的擴散模型中，得到所述擴散模型輸出的聚焦預(yù)測圖像的步驟之前，所述方法還包括：將獲取到的樣本圖像輸入待訓(xùn)練的初始模型中，以使所述初始模型對所述樣本圖像進行正向擴散處理，得到噪聲圖像；其中，所述樣本圖像聚焦于目標區(qū)域；根據(jù)所述初始模型對所述噪聲圖像進行反向擴散處理，得到去噪圖像和所述去噪圖像對應(yīng)的模型損失值；根據(jù)所述損失值對所述初始模型進行模型訓(xùn)練，得到所述擴散模型。

12、本技術(shù)第二方面提供了一種基于深度學(xué)習(xí)的變焦裝置，包括：聚焦預(yù)測模塊，用于將圖像采集設(shè)備采集到的目標圖像輸入預(yù)先訓(xùn)練的擴散模型中，得到所述擴散模型輸出的聚焦預(yù)測圖像；聚焦對比模塊，用于對比所述目標圖像中的目標聚焦區(qū)域與所述聚焦預(yù)測圖像中的預(yù)測聚焦區(qū)域，得到聚焦對比結(jié)果；變焦處理模塊，用于根據(jù)所述聚焦對比結(jié)果對所述圖像采集設(shè)備進行變焦處理。

13、本技術(shù)第三方面提供了一種電子設(shè)備，包括存儲器和處理器，處理器用于執(zhí)行存儲器中存儲的程序指令，以實現(xiàn)上述基于深度學(xué)習(xí)的變焦方法。

14、本技術(shù)第四方面提供了一種計算機可讀存儲介質(zhì)，其上存儲有程序指令，程序指令被處理器執(zhí)行時實現(xiàn)上述基于深度學(xué)習(xí)的變焦方法。

15、上述方案，通過將圖像采集設(shè)備采集到的目標圖像輸入預(yù)先訓(xùn)練的擴散模型中，由擴散模型對接收到的目標圖像進行擴散處理，得到擴散模型輸出的聚焦預(yù)測圖像；將目標圖像中的目標聚焦區(qū)域與聚焦預(yù)測圖像中的預(yù)測聚焦區(qū)域進行對比，得到聚焦對比結(jié)果，由此可以根據(jù)聚焦對比結(jié)果判斷圖像采集設(shè)備基于當(dāng)前變焦參數(shù)所采集到的目標圖像的目標區(qū)域是否與擴散模型預(yù)測的聚焦預(yù)測圖像中的預(yù)測聚焦區(qū)域相匹配；根據(jù)所述聚焦對比結(jié)果對所述圖像采集設(shè)備進行變焦處理，由此能夠?qū)崿F(xiàn)基于深度學(xué)習(xí)技術(shù)對圖像采集設(shè)備進行變焦，提高變焦過程的變焦準確性。

16、應(yīng)當(dāng)理解的是，以上的一般描述和后文的細節(jié)描述僅是示例性和解釋性的，而非限制本技術(shù)。