如何使用 tensorflow 旋转 3D 图像？

Question

我有一个图像张量：[batch, height, width, depth, 3] 和一个角度张量：[angleX, angleY, angleZ]。如何使用角度旋转图像？ tf.contrib.image.rotate 仅适用于 2D 图像。

顺便问一下，我可以使用“Y-rotate(X, angle)”作为损失函数吗？我不确定tensorflow是否可以计算旋转函数的梯度。

Answer 1

github 中有一个代码对旋转很有用。但是，它仅适用于 2D 图像。