无法使用Pythorch插入帧

Question

所以我试着测试在github页面中找到的一种算法，它可以插入帧。我在middlebury提供的数据集上测试该算法，链接：http://vision.middlebury.edu/flow/data/（我使用只有两个帧的彩色数据集）。这个错误也发生在约塞米蒂图像文件中）该算法对除最后一张外的所有图像都很有效。它会引发错误：

Traceback (most recent call last): 
File "run.py", line 257, in  
tensorInputFirst = torch.FloatTensor(numpy.rollaxis(numpy.asarray(PIL.Image.open(arguments_strFirst))[:,:,::-1], 2, 0).astype(numpy.float32) / 255.0) 
IndexError: too many indices for array

我知道这个错误意味着什么，一个不在列表中的索引，但我不知道它与帧插值的关系，以及它为什么会发生。有人能帮忙吗。在

我在github上找到的代码：

^{pr2}$

更新

我找到了错误的原因。这是因为图像是灰度的，而我测试的其他图像是彩色的，因为彩色的是24位深度，而灰度是8位深度。我如何修改代码，使其在灰度图像上工作。在