python - 有没有一种在 keras 中应用径向平均值的有效方法？

Question

我想在 keras 管道的末端应用径向平均值。

在倒数第二步，我有一个大小为 nx n 的图像。然后我想将此 nxn 图像映射到 1 xn/2 向量，其中向量 [x] = mean(image(radialPosition = x))。即我想平均距离图像中心距离 X 的所有点，并将其设置为输出 [x]。我们可以假设n是奇数，所以中心点是一个点。

我考虑过遍历所有半径，并选择所需的索引，以及图像和多个“平均”矩阵之间的点积，但这些似乎都没有计算效率。

有没有更好的方法来做到这一点？

标签： pythontensorflowmachine-learningkeras