keleeee 2022-10-07 16:54 采纳率: 60%
浏览 51
已结题

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

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

img

  • 写回答

3条回答 默认 最新

  • lzl2040 新星创作者: 人工智能技术领域 2022-10-07 17:02
    关注

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

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(2条)

报告相同问题？

问题事件

  • 系统已结题 10月16日
  • 已采纳回答 10月8日
  • 赞助了问题酬金15元 10月7日
  • 创建了问题 10月7日