ValueError: operands could not be broadcast together with shapes (418,558,3) (418,558)

这个错误通常是由于两个数组的形状不匹配导致的。具体地说，它表示一些尝试广播（broadcasting）两个数组的操作失败。

在你的代码中，可能是由于两个数组的形状不匹配导致了这个错误。你可以尝试检查一下数组的形状，并确保它们在进行操作时具有相同的形状。

例如，如果你正在尝试将一个形状为(418, 558, 3)的三维数组和一个形状为(418, 558)的二维数组相加，那么就会出现这个错误。在这种情况下，你可以使用numpy库的expand_dims()函数将二维数组的形状扩展为(418, 558, 1)，以便它可以与三维数组进行广播。

下面是一个示例代码，展示了如何使用expand_dims()函数来扩展数组的形状：

import numpy as np

# 生成两个形状分别为(418, 558, 3)和(418, 558)的数组
a = np.random.rand(418, 558, 3)
b = np.random.rand(418, 558)

# 将b的形状扩展为(418, 558, 1)
b = np.expand_dims(b, axis=2)

# 进行数组相加
c = a + b

在这个示例代码中，我们生成了两个随机数组a和b，其中a的形状为(418, 558, 3)，b的形状为(418, 558)。然后，我们使用expand_dims()函数将b的形状扩展为(418, 558, 1)，以便它可以与a进行广播。最后，我们将a和b相加，得到了一个形状为(418, 558, 3)的数组c。