如何使用 RGB 图像阵列的 PyTorch 张量在 colab 中显示视频？

如何解决如何使用 RGB 图像阵列的 PyTorch 张量在 colab 中显示视频？

我有一个形状为 (125,3,128,128) 的张量：

125 帧
3 通道 (RGB)
每帧 128 x 128 大小。
张量中的值在 [0,1] 范围内。

我想显示这 125 帧的视频，在 Google Colab 中使用 Pytorch。我该怎么做？

解决方法

在 Colab 中启用内联动画的一种方法是使用 jshtml：

from matplotlib import rc
rc('animation',html='jshtml')

启用此功能后，您可以绘制动画 like so（请注意，您需要 permute 图像张量才能将它们设为 PIL/matplotlib 格式）：

import matplotlib.pyplot as plt
import matplotlib.animation as animation

fig,ax = plt.subplots()

imgs = torch.rand(10,3,128,128)
imgs = imgs.permute(0,2,1) # Permuting to (Bx)HxWxC format
frames = [[ax.imshow(imgs[i])] for i in range(len(imgs))]

ani = animation.ArtistAnimation(fig,frames)
ani