用于進行流體模擬的分層神經(jīng)網(wǎng)絡(luò)的訓練方法及裝置

基本信息

申請?zhí)?/td> CN202210346497.7 申請日 -
公開(公告)號 CN114662397A 公開(公告)日 2022-06-24
申請公布號 CN114662397A 申請公布日 2022-06-24
分類號 G06F30/27(2020.01)I;G06T13/00(2011.01)I;G06K9/62(2022.01)I;G06V10/80(2022.01)I;G06V10/82(2022.01)I;G06N3/08(2006.01)I;G06F113/08(2020.01)N 分類 計算;推算;計數(shù);
發(fā)明人 樊思明;樸鏡潭;林君儀;錢晨 申請(專利權(quán))人 商湯集團有限公司
代理機構(gòu) 北京中知恒瑞知識產(chǎn)權(quán)代理事務(wù)所(普通合伙) 代理人 -
地址 中國香港新界沙田香港科學園科技大道西一號核心大樓第二座2樓226-230室
法律狀態(tài) -

摘要

摘要 本公開提供了一種用于進行流體模擬的分層神經(jīng)網(wǎng)絡(luò)的訓練方法及裝置,包括:獲取樣本視頻中的多幀樣本視頻幀,以及所述樣本視頻中與所述多幀樣本視頻幀對應(yīng)的監(jiān)督視頻幀,以及所述多幀樣本視頻幀分別與所述監(jiān)督視頻幀之間的光流信息;將所述多幀樣本視頻幀輸入至待訓練的分層神經(jīng)網(wǎng)絡(luò)中,確定所述多幀樣本視頻幀分別對應(yīng)的背景特征信息和流體特征信息;基于各樣本視頻幀對應(yīng)的光流信息調(diào)整各樣本視頻幀對應(yīng)的流體特征信息,得到調(diào)整流體特征信息;基于所述背景特征信息和所述調(diào)整流體特征信息構(gòu)建預(yù)測視頻幀,并基于所述預(yù)測視頻幀和所述監(jiān)督視頻幀對所述待訓練的分層神經(jīng)網(wǎng)絡(luò)進行訓練。