用改进后模型的权重进行detect.py时报错

Question

用改进后模型的权重进行detect.py时报错，有偿求如何解决这个问题

Answer 1

把你输入conv2d函数的数据用这段代码处理一下
data = data.detach().cpu().numpy()
data = torch.FloatTensor(data)
这个data就是你的数据
FloatTensor如果不行的话改成:
data=torch.HalfTensor(data)