作者您好，运行main_infer.py时遇到 RuntimeError: Expected 3D (unbatched) or 4D (batched) input to conv2d, but got input of size: [1, 6, 6, 512, 512] 问题，定位到第128行代码：img_tensor = torch.stack(all_data, dim=0).unsqueeze(0) #.cuda() 输出的形状是5d，请问这种数据如何处理？ 期待解答，谢谢~
作者您好，运行main_infer.py时遇到
RuntimeError: Expected 3D (unbatched) or 4D (batched) input to conv2d, but got input of size: [1, 6, 6, 512, 512]
问题，定位到第128行代码：img_tensor = torch.stack(all_data, dim=0).unsqueeze(0) #.cuda()
输出的形状是5d，请问这种数据如何处理？
期待解答，谢谢~