[發(fā)明專利]數(shù)據(jù)處理方法、裝置、電子設(shè)備及存儲介質(zhì)

在審

• 申請?zhí)枺?/td>: 202010886520.2
• 申請日: 2020-08-28
• 公開（公告）號: CN111899150A
• 公開（公告）日: 2020-11-06
• 發(fā)明（設(shè)計）人: 譚志鵬
• 申請（專利權(quán)）人: OPPO廣東移動通信有限公司
• 主分類號: G06T1/20
• 分類號: G06T1/20;G06N5/04
• 代理公司: 深圳市智圈知識產(chǎn)權(quán)代理事務(wù)所(普通合伙) 44351
• 代理人: 呂靜

1.一種數(shù)據(jù)處理方法，其特征在于，應(yīng)用于電子設(shè)備的圖形處理器GPU，所述電子設(shè)備還包括中央處理器CPU，所述方法包括：

響應(yīng)所述CPU發(fā)起的數(shù)據(jù)處理指令，進行深度學(xué)習(xí)框架運行前的環(huán)境初始化；

運行所述深度學(xué)習(xí)框架，并接收所述CPU傳輸?shù)拇幚頂?shù)據(jù)；

對所述待處理數(shù)據(jù)進行所述深度學(xué)習(xí)框架中的預(yù)處理；

對預(yù)處理后的待處理數(shù)據(jù)進行所述深度學(xué)習(xí)框架中深度學(xué)習(xí)模型的推理計算；

對所述推理計算獲得的結(jié)果進行所述深度學(xué)習(xí)框架中的后處理，并將獲得的后處理結(jié)果傳輸至所述CPU。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述對所述待處理數(shù)據(jù)進行所述深度學(xué)習(xí)框架中的預(yù)處理，包括：

判斷所述深度學(xué)習(xí)框架中預(yù)先定義的第一函數(shù)指針是否被注冊；

如果所述第一函數(shù)指針被注冊，則基于所述第一函數(shù)指針，調(diào)用預(yù)處理函數(shù)；

基于所述預(yù)處理函數(shù)對所述待處理數(shù)據(jù)進行預(yù)處理。

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，在所述判斷所述深度學(xué)習(xí)框架中預(yù)先定義的第一函數(shù)指針是否被注冊之后，所述方法還包括：

如果所述第一函數(shù)指針未被注冊，則對所述待處理數(shù)據(jù)進行所述深度學(xué)習(xí)模型的推理計算；

對所述推理計算獲得的結(jié)果進行所述深度學(xué)習(xí)框架中的后處理，并將獲得的后處理結(jié)果傳輸至所述CPU。

4.根據(jù)權(quán)利要求1或3所述的方法，其特征在于，在所述對所述推理計算獲得的結(jié)果進行所述深度學(xué)習(xí)框架中的后處理，并將獲得的后處理結(jié)果傳輸至所述CPU之前，所述方法還包括：

判斷所述深度學(xué)習(xí)框架中預(yù)先定義的第二函數(shù)指針是否被注冊；

所述對所述推理計算獲得的結(jié)果進行所述深度學(xué)習(xí)框架中的后處理，并將獲得的后處理結(jié)果傳輸至所述CPU，包括：

如果所述第二函數(shù)指針被注冊，則基于所述第二函數(shù)指針，調(diào)用后處理函數(shù)；

基于所述后處理函數(shù)對所述推理計算獲得的結(jié)果進行后處理，并將獲得的后處理結(jié)果傳輸至所述CPU。

5.根據(jù)權(quán)利要求4所述的方法，其特征在于，在所述判斷所述深度學(xué)習(xí)框架中預(yù)先定義的第二函數(shù)指針是否被注冊之后，所述方法還包括：

如果所述第二函數(shù)指針未被注冊，則將所述推理計算獲得的結(jié)果傳輸至所述CPU。

6.根據(jù)權(quán)利要求1或3所述的方法，其特征在于，在所述對所述推理計算獲得的結(jié)果進行所述深度學(xué)習(xí)框架中的后處理，并將獲得的后處理結(jié)果傳輸至所述CPU之后，所述方法還包括：

將所述環(huán)境初始化后的參數(shù)進行還原。

7.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述進行深度學(xué)習(xí)框架運行前的環(huán)境初始化，包括：

分配用于運行所述深度學(xué)習(xí)框架的GPU資源；

創(chuàng)建所述深度學(xué)習(xí)框架的運行環(huán)境。

8.一種數(shù)據(jù)處理裝置，其特征在于，應(yīng)用于電子設(shè)備的圖形處理器GPU，所述電子設(shè)備還包括中央處理器CPU，所述裝置包括：初始化模塊、數(shù)據(jù)接收模塊、預(yù)處理模塊、推理模塊以及后處理模塊，其中，

所述初始化模塊用于響應(yīng)所述CPU發(fā)起的數(shù)據(jù)處理指令，進行深度學(xué)習(xí)框架運行前的環(huán)境初始化；

所述數(shù)據(jù)接收模塊用于運行所述深度學(xué)習(xí)框架，并接收所述CPU傳輸?shù)拇幚頂?shù)據(jù)；

所述預(yù)處理模塊用于對所述待處理數(shù)據(jù)進行所述深度學(xué)習(xí)框架中的預(yù)處理；

所述推理模塊用于對預(yù)處理后的待處理數(shù)據(jù)進行所述深度學(xué)習(xí)框架中深度學(xué)習(xí)模型的推理計算；

所述后處理模塊用于對所述推理計算獲得的結(jié)果進行所述深度學(xué)習(xí)框架中的后處理，并將獲得的后處理結(jié)果傳輸至所述CPU。

9.一種電子設(shè)備，其特征在于，包括：

一個或多個處理器，所述處理器包括中央處理器CPU及圖形處理器GPU；

存儲器；

一個或多個應(yīng)用程序，其中所述一個或多個應(yīng)用程序被存儲在所述存儲器中并被配置為由所述一個或多個處理器執(zhí)行，所述一個或多個程序配置用于執(zhí)行如權(quán)利要求1-7任一項所述的方法。

10.一種計算機可讀取存儲介質(zhì)，其特征在于，所述計算機可讀取存儲介質(zhì)中存儲有程序代碼，所述程序代碼可被處理器調(diào)用執(zhí)行如權(quán)利要求1-7任一項所述的方法。