yolov5 导出torchscript报错：RuntimeError: Input type (torch.cuda.FloatTensor) and weight type

最新推荐文章于 2024-03-27 17:50:22 发布
最新推荐文章于 2024-03-27 17:50:22 发布
阅读量1.9k 收藏 4
点赞数 1
文章标签： pytorch 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/m0_52948781/article/details/124671318
版权

github ultralytics的yolov5 v6.0版本代码导出torchscript导出成功了但是detect.py推理的时候报RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

这里有两种解决思路

1. 将输入input和weight统一在一个设备上（我采用的这个，比较简单粗暴）

排查代码可以发现，在detect.py中有个DetectMultiBanckend这是用于载入不同类型的模型的类，该类定义在 根目录/models/common.py中，查看该函数对应的 jit（即torchscript） 处代码

 可见这里少了一个参数，map_location=device，加上即可

 

2. 参考(97条消息) Pytorch避坑之：RuntimeError: Input type(torch.cuda.FloatTensor) and weight type(torch.FloatTensor) shoul_暖仔会飞的博客-CSDN博客

m0_52948781
关注
  • 1
    点赞
  • 踩
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Vue项目报错：Uncaught SyntaxError: Unexpected token <
10-17
主要介绍了Vue项目报错：Uncaught SyntaxError: Unexpected token <，在引入第三方依赖的 JS 文件时，遇到的一个问题，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
深度学习常见错误及解决方案：RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)
神马都会亿点点的毛毛张
03-03 542
错误内容就在类型不匹配，根据报错内容可以看出Input type为torch.FloatTensor（CPU数据类型），而weight type(即网络权重参数这些)为torch.cuda.FloatTensor（GPU数据类型）。将输入类型转变为GPU类型或者映射到CPU类型。
PyTorch 深度学习框架 torch、torchvision、libtorch 库，以及将 PyTorch 模型导出为 TorchScript（.ts 文件）
最新发布
没有余地 EliasJie
03-27 817
仅供本人记录查阅使用
目标检测实战(六): 使用YOLOv8完成对图像的目标检测任务（从数据准备到训练测试部署的完整流程）
源的博客
09-08 1万+
目标检测（Object Detection）是计算机视觉领域的一项重要技术，旨在识别图像或视频中的特定目标并确定其位置。通过训练深度学习模型，如卷积神经网络（CNN），可以实现对各种目标的精确检测。常见的目标检测任务包括：人脸检测、行人检测、车辆检测等。目标检测在安防监控、自动驾驶、智能零售等领域具有广泛应用前景。
在cpu还是gpu上运算的问题RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)
qq_53821866的博客
11-28 89
【代码】在cpu还是gpu上运算的问题RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)
复刻yolo系列时出现的BUG及解决方法
m0_59304732的博客
11-24 6338
一、ValueError: too many values to unpack (expected 2) 二、RuntimeError: result type Float can't be cast to the desired output type __int64 三、AssertionError: Torch not compiled with CUDA enabled 四、RuntimeError: "slow_conv2d_cpu" not implemented for 'Half'
记录一下yolov5这个报错：Inplace update to inference tensor outside InferenceMode is not allowed.
qq_38077313的博客
02-06 2710
yolov5 报错处理
YOLOv5接口报错
人纇捕完計劃
05-17 798
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be the same 个人分析报错原因是在使用单GPU的时候Yolov5会开启半精度，但是我在数据处理的时候漏掉了这种情况。 临时解决方式： 由于时间有限，暂时把单GPU开启半精度的功能注释掉了 更好的方式： 在数据处理阶段增加使用半精度模式时img = img.half() ...
运行export.py转化.pt模型时报错：Inplace update to inference tensor outside Inference，修改torch.inference最低指定版本即可
qwerdf6660的博客
12-09 773
然后修改minmum:后面的torch版本（图中绿色数字），我当前环境的torch是1.9.0版本的，所以修改大于这个版本即可。这里是指将生成的best.pt模型进行转化，具体要根据自己best.pt路径来写，其余部分保持不变。
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.cuda.HalfTensor) should be.
XC___XC的博客
10-16 1万+
device 首先是查看输入的两个数据的device是否一致，出现这种情况的原因可能是因为device不同导致的，可能在不同的GPU上。 device = torch.device("cuda" if torch.cuda.is_available() else "cpu") data.to(device) 通过这个设定一个device，然后将data转到device中，然后debug查看数据的device是否相同。 type 在我这里，我的两个输入数据是在同一个cuda中，但是还是出错了，通过查看数据
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the
qq_45160840的博客
11-26 747
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the
基于YOLOv5和onnxruntime C++实现目标检测（源码+说明文档）.rar
04-15
1、资源内容：基于yolov5检测onnxruntime c++（完整源码+说明文档+数据）.rar 2、代码特点：参数化编程、参数可方便更改、代码编程思路清晰、注释明细。 3、适用对象：计算机，电子信息工程、数学等专业的大学生课程...
yolov5-onnx-tensorrt:如何使用 TensorRT 运行 yolov5 模型
05-29
yolov5-onnx-张量 此 Repos 包含如何使用 TensorRT 运行 yolov5 模型。 Pytorch 实现是 。 将 pytorch 转换为 onnx 和 tensorrt yolov5 模型以在 Jetson AGX Xavier 上运行。 支持推断图像。 支持同时推断多幅图像。...
Python RuntimeError: thread.__init__() not called解决方法
09-22
主要介绍了Python RuntimeError: thread.__init__() not called解决方法,需要的朋友可以参考下
YOLOv8 深度详解！一文看懂，快速上手
热门推荐
qq_39967751的博客
01-12 2万+
本文详细分析和总结了最新的 YOLOv8 算法，从整体设计到模型结构、Loss 计算、训练数据增强、训练策略和推理过程进行了详细的说明，并提供了大量的示意图供大家方便理解。
Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
y473459100的博客
03-19 9940
使用pytorch测试模型时，报如下错误： RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same 报错信息意思是说，我们输入的数据类型与网络参数的类型不符。 Input type的类型时torch.cuda.FloatTensor（GPU数据类型） weight type（即net.parameters）为torch.FloatTensor
新手避坑：RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be
super199812的博客
10-24 3469
新手Pytorch避坑
[报错]-RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) shoul
Y_1239的博客
03-23 1万+
模型输入的数据类型要与模型参数的数据类型一致。
runtimeerror: input type (torch.cuda.floattensor) and weight type (torch.flo
09-18
此错误是由于输入类型(torch.cuda.FloatTensor)和权重类型(torch.FloatTensor)不匹配所致。PyTorch中的张量类型包括CPU上的FloatTensor和GPU上的CUDA FloatTensor。在此情况下，输入张量被放置在GPU上，而权重张量却被放置在CPU上，因此引发了RuntimeError。 要解决这个问题，可以尝试以下几种方法： 1. 确保输入和权重张量使用相同的设备进行操作。如果想要在GPU上进行操作，需要将权重张量发送到GPU上。 2. 使用.to()方法将张量转移到相应的设备上。例如，可以使用input_tensor.to('cuda')将输入张量转移到GPU上。 3. 如果权重张量已经在GPU上但类型不匹配，可以使用.to()方法更改权重张量类型。例如，可以使用weight_tensor.to(torch.cuda.FloatTensor)将权重张量类型更改为与输入张量相匹配。 需要注意的是，对于.to()方法，可以使用字符串'cuda'或torch.device对象引用CUDA设备，并使用'torch.device('cuda')'将权重张量移动到CUDA设备上。 总之，通过将输入和权重张量放置在相同的设备上并确保它们的类型匹配，可以解决RuntimeError。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值