我在执行以下代码时收到一个错误：

最后一行引发了”TypeError: forward() takes 2 positional arguments but 5 were given”错误。img的大小是[1, 3, 960, 1280]。如果我在传入模型之前在img周围添加方括号（output = model([img])），我会得到”ValueError: images is expected to be a list of 3d tensors of shape [C, H, W], got torch.Size([1, 3, 960, 1280])”。然后，如果我使用”img.view(3, 960, 1280)”，我又会得到最初的错误。

这个问题有什么解决方案吗？谢谢。

回答：

我不确定你用这行代码想要做什么：

model.roi_heads = nn.Sequential()

这一行代码引起了问题，我在没有这行代码的情况下运行了你的代码片段，一切正常（除了来自模型使用的几个关于废弃的警告，但输出看起来是合理的）