model(tensor_image)输出不了任何内容

Question

model = torch.load("u2net.pth", map_location=torch.device('cpu'))没问题
tensor_image也没问题，但是model(tensor_image)却得不到任何输出

Answer 1

基于new BIng和ChatGPT的回答：
可能是因为模型在GPU上被训练并保存，但是你却在CPU上加载模型和输入数据。尝试使用以下代码将模型转换为CPU模式：

model = torch.load("u2net.pth", map_location=torch.device('cpu'))
model.eval()

另外，检查一下模型的输出形状是否正确，应该是一个四维张量，形状为(batch_size, num_channels, height, width)。如果输出形状不正确，可能需要调整模型或输入数据的形状。