如何使用 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 图像。