[发明专利]图像处理方法和装置、电子设备、存储介质、程序产品

说明书

本申请实施例公开了一种图像处理方法和装置、电子设备、存储介质、程序产品，其中，方法包括：通过至少两个第一胶囊对待处理图像进行特征提取处理，得到至少两个第一特征数据，其中，所述至少两个特征数据中每个特征数据包含多个向量；基于所述至少两个第一特征数据，确定第二胶囊的主特征数据和侧特征数据，其中，所述第二胶囊所属的网络层位于所述至少两个第一胶囊的网络层之后；通过所述第二胶囊对所述主特征数据和所述侧特征数据进行处理，得到第二特征数据；基于所述第二特征数据，得到图像处理结果，降低了图像处理的复杂度，提高图像处理性能。

技术领域

本申请涉及计算机视觉技术，尤其是一种图像处理方法和装置、电子设备、存储介质、程序产品。

背景技术

神经网络在图像处理方面已取得了广泛应用和发展。近期，研究者提出了将胶囊网络应用于图像处理，在胶囊网络中，神经元被胶囊所替代，胶囊可以用向量来表示，与普通的神经网络相比，胶囊网络具有更好的图像处理性能。

发明内容

本申请实施例提供了一种图像处理技术。

根据本申请实施例的一个方面，提供了一种图像处理方法，包括：

通过至少两个第一胶囊对待处理图像进行特征提取处理，得到至少两个第一特征数据，其中，所述至少两个特征数据中每个特征数据包含多个向量；

基于所述至少两个第一特征数据，确定第二胶囊的主特征数据和侧特征数据，其中，所述第二胶囊所属的网络层位于所述至少两个第一胶囊的网络层之后；

通过所述第二胶囊对所述主特征数据和所述侧特征数据进行处理，得到第二特征数据；

基于所述第二特征数据，得到图像处理结果。

可选地，所述基于所述至少两个第一特征数据，确定第二胶囊的主特征数据和侧特征数据，包括：

将所述至少两个第一胶囊中位置与所述第二胶囊对应的主胶囊得到的第一特征数据作为所述第二胶囊的主特征数据；

基于所述至少两个第一胶囊中除所述主胶囊之外的至少一个侧胶囊得到的第一特征数据，确定所述第二胶囊的侧特征数据。

可选地，所述基于所述至少两个第一胶囊中除所述主胶囊之外的至少一个侧胶囊得到的第一特征数据，确定所述第二胶囊的侧特征数据，包括：

对所述至少一个侧胶囊得到的第一特征数据进行卷积处理，获得所述第二胶囊的侧特征数据。

可选地，所述通过所述第二胶囊对所述主特征数据和所述侧特征数据进行处理，得到第二特征数据，包括：

基于所述主特征数据和所述侧特征数据，确定所述第二胶囊的输入数据；

通过所述第二胶囊对所述输入数据进行处理，获得第二特征数据。

可选地，所述基于所述主特征数据和所述侧特征数据，确定所述第二胶囊的输入数据，包括：

基于所述主特征数据、所述主特征数据的第一权重、所述侧特征数据和所述侧特征数据的第二权重，得到所述第二胶囊的输入数据。

可选地，所述第一权重和所述第二权重是通过训练得到的。

可选地，所述方法利用胶囊网络实现，所述胶囊网络包括至少两个网络层，每个所述网络层包括至少一个胶囊；

所述通过至少两个第一胶囊对待处理图像进行特征提取处理，得到至少两个第一特征数据之前，还包括：

将样本图像输入所述胶囊网络；

基于所述至少两个网络层中两个所述网络层输出的两个特征图之间的特征距离，获得第一损失；

基于所述第一损失训练所述胶囊网络。