[发明专利]扩散模型处理方法及装置、图片处理方法及装置

权利要求书

1.一种扩散模型处理方法，包括：

确定扩散模型的时间步集合、以及所述时间步集合对应的时间步区间；

从所述时间步集合中确定第一时间步，并根据所述时间步区间确定所述第一时间步对应的目标时间步，其中，所述第一时间步为所述时间步集合中的任一时间步；

将所述第一时间步对应的加噪图片以及所述目标时间步，输入扩散模型，获得所述加噪图片对应的预测噪声；

根据所述加噪图片对应的目标噪声以及所述预测噪声，处理所述扩散模型。

2.根据权利要求1所述的扩散模型处理方法，所述将所述第一时间步对应的加噪图片以及所述目标时间步，输入扩散模型，获得所述加噪图片对应的预测噪声之前，还包括：

确定初始图片以及、所述第一时间步对应的目标噪声；

根据所述初始图片以及所述目标噪声，确定所述第一时间步对应的加噪图片以及、所述加噪图片对应的目标噪声。

3.根据权利要求1所述的扩散模型处理方法，所述确定扩散模型的时间步集合、以及所述时间步集合对应的时间步区间，包括：

确定扩散模型的时间步集合，并根据预设划分条件对所述时间步集合中的时间步进行区间划分，获得所述时间步集合对应的时间步区间。

4.根据权利要求1所述的扩散模型处理方法，所述根据所述时间步区间确定所述第一时间步对应的目标时间步，包括：

确定所述时间步区间的区间端点，根据所述区间端点确定所述第一时间步对应的目标时间步。

5.根据权利要求4所述的扩散模型处理方法，所述确定所述时间步区间的区间端点，根据所述区间端点确定所述第一时间步对应的目标时间步，包括：

确定所述时间步区间的区间左端点，并将所述区间左端点确定为所述第一时间步对应的目标时间步，其中，所述时间步区间的区间右端点为下一时间步区间包括的左端点。

6.根据权利要求4所述的扩散模型处理方法，所述确定所述时间步区间的区间端点，根据所述区间端点确定所述第一时间步对应的时间步，包括：

确定所述时间步区间的区间右端点，并将所述区间右端点确定为所述第一时间步对应的目标时间步，其中，所述时间步区间的区间左端点为上一时间步区间包括的右端点。

7.根据权利要求1所述的扩散模型处理方法，所述根据所述加噪图片对应的目标噪声以及所述预测噪声，处理所述扩散模型，包括：

根据所述加噪图片对应的目标噪声以及所述预测噪声计算噪声损失函数，并根据所述噪声损失函数调整所述扩散模型的网络参数，并在满足预设训练结束条件的情况下，获得所述扩散模型。

8.根据权利要求1-7任意一项所述的扩散模型处理方法，所述根据所述加噪图片对应的目标噪声以及所述预测噪声，处理所述扩散模型之后，还包括：

确定目标加噪图片，将所述目标加噪图片输入扩散模型，获得所述目标加噪图片对应的预测噪声；

根据所述目标加噪图片以及所述目标加噪图片对应的预测噪声，确定去噪后的目标图片。

9.根据权利要求8所述的扩散模型处理方法，所述确定目标加噪图片，包括：

确定加噪的视频帧集合，并将所述视频帧集合中的任一视频帧，确定为目标加噪图片。

10.一种扩散模型处理装置，包括：

区间划分模块，被配置为确定扩散模型的时间步集合、以及所述时间步集合对应的时间步区间；

目标时间步确定模块，被配置为从所述时间步集合中确定第一时间步，并根据所述时间步区间确定所述第一时间步对应的目标时间步，其中，所述第一时间步为所述时间步集合中的任一时间步；

第一模型预测模块，被配置为将所述第一时间步对应的加噪图片以及所述目标时间步，输入扩散模型，获得所述加噪图片对应的预测噪声；

模型处理模块，被配置为根据所述加噪图片对应的目标噪声以及所述预测噪声，处理所述扩散模型。