Closed SunHui1216 closed 3 months ago
您好,在该py文件下的测试代码如下: def unit_test(): import numpy as np x = torch.tensor(np.random.rand(2,4,480,640).astype(np.float32)) model = FusionNet(output=1) y = model(x) print('output shape:', y.shape) assert y.shape == (2,1,480,640), 'output shape (2,1,480,640) is expected!' print('test ok!') 在运行后会报错:TypeError: forward() missing 1 required positional argument: 'image_ir' 请问原因是否为您这里的测试输入是错误的? 感谢回答!
非常抱歉 这个测试代码是我们前一个版本使用的 要对现在的FusionNet进行测试你需要定义两个输入 即一个红外图像和一个可见光图像。
感谢您的回复!
您好,在该py文件下的测试代码如下: def unit_test(): import numpy as np x = torch.tensor(np.random.rand(2,4,480,640).astype(np.float32)) model = FusionNet(output=1) y = model(x) print('output shape:', y.shape) assert y.shape == (2,1,480,640), 'output shape (2,1,480,640) is expected!' print('test ok!') 在运行后会报错:TypeError: forward() missing 1 required positional argument: 'image_ir' 请问原因是否为您这里的测试输入是错误的? 感谢回答!