[发明专利]一种图像重构方法及装置

权利要求书

1.一种图像重构方法，其特征在于，所述方法包括：

采用不同的分割方式对待处理图像进行分割，得到不同的子图像集，其中，每一子图像集中包含：采用一种分割方式对待处理图像进行分割得到的所有子图像；

针对每一子图像集，采用目标分割方式对应的图像重构处理的方式对该子图像集中包含的各个子图像分别进行图像重构处理，得到各个子图像的子图像重构结果，其中，所述目标分割方式为：获得该子图像集中包含的子图像时使用的分割方式；

针对每一子图像集，对该子图像集中包含的各个子图像的子图像重构结果进行合并，得到与所述待处理图像尺寸相同、与目标分割方式相对应的重构图像；

基于各个预设的注意力权重，对各个重构图像进行融合，得到最终的图像重构结果，其中，各个注意力权重与各个分割方式一一对应，每一注意力权重表示：该注意力权重对应的分割方式对应的重构图像的预测清晰度。

2.根据权利要求1所述的方法，其特征在于，所述采用目标分割方式对应的图像重构处理的方式对该子图像集中包含的各个子图像分别进行图像重构处理，包括：

针对该子图像集中的每一子图像，通过以下方式对该子图像进行图像重构处理：

对该子图像进行通道划分，得到不同的通道图像；

分别对各个通道图像进行与目标分割方式相对应的全连接处理，得到第一结果；

对各个第一结果进行融合，得到与该子图像相对应的第二结果；

对所述第二结果进行与目标分割方式相对应的重构处理，得到该子图像的子图像重构结果。

3.根据权利要求2所述的方法，其特征在于，所述分别对各个通道图像进行与目标分割方式相对应的全连接处理，得到第一结果，包括：

分别将各个通道图像输入预先训练好的、目标分割方式对应的全连接层，将输出结果作为第一结果；

所述对所述第二结果进行与目标分割方式相对应的重构处理，得到该子图像的子图像重构结果，包括：

将所述第二结果输入预先训练好的、所述目标分割方式对应的重构模型，将输出结果作为该子图像的子图像重构结果。

4.根据权利要求3所述的方法，其特征在于，针对每一分割方式，通过以下方式对预设全连接层进行训练，得到该分割方式对应的全连接层，并对预设模型进行训练，得到该分割方式对应的重构模型：

采用该分割方式对样本图像进行分割，得到样本子图像；

对各个样本子图像进行图像重构处理，得到样本对照结果；

分别对各个样本子图像进行通道划分，得到各个样本子图像的样本通道图像；

将各个样本通道图像输入所述预设全连接层，将输出结果作为第一样本结果；

针对每一样本子图像，对基于该样本子图像生成的各个第一样本结果进行融合，得到该样本子图像对应的第二样本结果；

针对每一样本子图像，将该样本子图像对应的第二样本结果输入所述预设模型，将输出结果作为该样本子图像的样本重构结果；

基于各个样本子图像的样本对照结果和样本重构结果，计算该分割方式对应的子损失；

基于各个分割方式对应的子损失，计算总损失；

基于所述子损失调整所述预设全连接层的参数，并基于所述总损失调整所述预设模型的参数；

返回执行所述采用该分割方式对样本图像进行分割，得到样本子图像的步骤，直至达到预设的训练终止条件，将参数调整后的预设全连接层确定为该分割方式对应的全连接层，将参数调整后的预设模型确定为该分割方式对应的重构模型。

5.根据权利要求4所述的方法，其特征在于，所述基于各个分割方式对应的子损失，计算总损失，包括：

基于各个子损失，计算各个分割方式当前对应的注意力权重；

基于各个子损失和各个注意力权重，计算总损失。

6.根据权利要求4所述的方法，其特征在于，所述基于所述子损失调整所述预设全连接层的参数，包括：

基于该分割方式对应的子损失计算更新系数，所述更新系数表示：调整所述预设全连接层的参数时的调整幅度；

基于所述更新系数与所述子损失，调整所述预设全连接层的参数。