yolov5 export.py文件导出模型有误:TorchScript: export failure: save(): incompatible function arguments.

最新推荐文章于 2024-08-23 02:39:12 发布
最新推荐文章于 2024-08-23 02:39:12 发布
阅读量2k 收藏 6
点赞数 8
文章标签： Yolov5 TorchScript 模型导出 错误修复 PyTorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/qq_42607586/article/details/120527948
版权

yolov5 export.py文件导出模型有误:TorchScript: export failure: save(): incompatible function arguments. The following argument types are supported:

  1. (self: torch._C.ScriptModule, filename: str, _extra_files: Dict[str, str] = {}) -> None

运行yolov5 export.py文件报错：

TorchScript: export failure: save(): incompatible function arguments. The following argument types are supported:

  1. (self: torch._C.ScriptModule, filename: str, _extra_files: Dict[str, str] = {}) -> None Invoked with: <torch._C.ScriptModule object at 0x7fc81d2dd538>

  2. 解决方法：

  3. 在f = file.with_suffix(’.torchscript.pt’)后加上这句代码
    f = str(f)
    亲测可用可以在插入图片描述

  4. 参考链接：

链接:https://github.com/ultralytics/yolov5/issues/4352

酥酥禾
关注
YOLOv5的Tricks | 【Trick14】YOLOv5的val.py脚本的解析
Clichong
08-12 1万+
如有问题，恳请指出。这篇可能是这个系列最后的一篇了，最后把yolov5的验证过程大致的再介绍介绍，基本上把yolov5的全部内容就稍微过了一遍了，也是我自己对这个项目学习的结束。（补充一下，这里我介绍的yolov5-6.0版本的代码）这个脚本主要分为是三个部分：主体代码运行部分 + 指标计算部分 + 绘图部分，这里就主要介绍前两个部分的内容。细节比较多，比上一篇的detect.py脚本复杂很多，这里写得困难逻辑也不会很清晰，简单做个记录，方便日后自己回忆。在训练阶段每个batch训练结束后，都会调用一次va
YOLOV5测试代码test.py注释与解析
Q1u1NG的博客
07-20 3万+
YOLOV5测试代码test.py注释与解析测试参数以及main函数解析test函数解析 本文主要对ultralytics\yolov5的训练代码train.py的解析，由于yolov5还在开发当中，平常多多少少都会修复一些bug或者有一些代码和功能的更新，但基本上不会有很大的改动，故以下注释与解析都是适用的；当然如果有大改动，笔者也会更新注释。 测试参数以及main函数解析 if __name__ == '__main__': """ opt参数详解 weights:测试的模型权重
yolov5模型直接导出torchscript格式、onnx格式
2301_77176270的博客
09-23 2033
最近在学习训练yolov5模型，但在导出onnx格式文件总是出错，参考了很多文章都是采用直接运行export.py文件，但我确只导出了torchscript格式文件。我运行以上代码，导出的只有 torchscript格式文件，并没有其他人写的onnx格式文件，最后我上github的查看官方的文档，发现导出语句已经进行更改如下。成功解决问题导出了best.onnx文件！
怎样将YOLOv5部署到移动端 报错：name ‘SPPF‘ is not defined
qq_41799012的博客
12-20 5826
问题描述： ultralytics发布的yolov5 6.0版本中已经加入了export.py文件可以将训练好的模型.pt转换为.tflite格式的轻量化模型部署在Android端，但是注意6.0版本当前代码只能转换5.0版本的。 我在使用6.0版本的export.py代码时报错 name ‘SPPF’ is not defined，找了半天报错的位置才发现，6.0版本模型文件中model/commen.py文件夹中多了一个SPPF模块来代替5.0版本中的SPP模块，当然旧的SPP模块依旧还在，而在将模型转
Yolov5——使用export.py把pt格式模型输出为onnx模型
最新发布
myhfory的博客
08-23 2092
需要Python>=3.8.0，并且含有PyTorch>=1.8，安装好yolov5所需要的所有依赖。
记录一下yolov5这个报错：Inplace update to inference tensor outside InferenceMode is not allowed.
qq_38077313的博客
02-06 3325
yolov5 报错处理
Python3与torch(torchvision)版本冲突问题
caobo_0512的博客
07-03 5874
Python3与torch(torchvision)版本冲突问题 按照torchgu官网，https://pytorch.org/的方法进行安装，但出现的界面与以前的不太一样，语言选项已经没有python的版本了，然后安装的是cuda9.0，上边也没有的这个选项，可以选择9.2的版本。 安装完毕。导入torch正常，导入Torchvision时出现TypeError: _resolve_type_from_object(): incompatible function arguments. The foll
yolov5使用官方提供的export.py导出模型报错问题
athrunsunny的博客
10-20 9666
1、TorchScript: export failure: save(): incompatible function arguments. The following argument types are supported: (self: torch._C.ScriptModule, filename: str, _extra_files: Dict[str, str] = {}) -> None 解决办法： 在export_torchscript()函数中添加：f = st...
TypeError: _queue_reduction(): incompatible function arguments.
01-20
TypeError: _queue_reduction(): incompatible function arguments. The following argument types are supported: 1. (process_group: torch.distributed.ProcessGroup, grads_batch: List[List[at::Tensor]], ...
yolov5——detect.py代码【注释、详解、使用教程】
热门推荐
CharmsLUO的博客
03-12 8万+
yolov5——detect.py代码【注释、详解、使用教程】 根据目前的最新版本的yolov5代码做出注释和详解以及使用教程，对了目前已经是v6,不知道你看博客的时候是什么版本呢，总的来说越来越先进越来越完善，越来越适合无脑啊哈哈哈，没你说哈IIII detect.py代码注释详解 1. 函数parse_opt（） def parse_opt(): """ weights: 训练的权重路径,可以使用自己训练的权重,也可以使用官网提供的权重 默认官网的权重yolov5s.pt(yo
Cause: com.mysql.jdbc.exceptions.jdbc4.MySQLSyntaxErrorException: ….. this is incompatible with sq
12-14
3、Can’t connect to local MySQL server through socket ‘/Data/mydata/mysql.sock’ socket文件目录不对应导致的问题 4、今天要说的就是 没有打开only_full_group_by Cause:...
yolov5平台模型格式转换&安卓端部署完整记录
Satellite的博客
06-30 4079
yolov5模型转换及安卓端部署
转torchscript报错：Expected a value of type ‘Tensor (inferred)‘ for argument ‘scale‘ but instead found t
Teng的博客
10-18 1486
Expected a value of type 'Tensor (inferred)' for argument 'scale' but instead found type 'int'. Inferred 'scale' to be of type 'Tensor' because it was not annotated with an explicit type.
极简yolov5转torchscript
zzzhaowendao的博客
03-18 4727
最近在使用yolov5转torchscript模型做推理时遇到了这样一个问题： RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! 不是所有tensor都在同个设备上，但是检查了model的weights和input都是在一个设备上，就很疑惑，后来发现是yolov5推理时的detect 层的融框操作导致。
yolov5 导出torchscript报错：RuntimeError: Input type (torch.cuda.FloatTensor) and weight type
m0_52948781的博客
05-09 2074
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中有个Detec
yolov5-lite v1.5版本部署时遇到的一些问题
weixin_39743452的博客
04-10 299
由于在本地部署该项目时，即使安装了requirements上的库文件还是会遇到一些问题，可能是因为某些库版本过高的原因，因此在此记录一下。首先是笔者的Python版本和安装的库的版本。这里是numpy版本过高的原因，降到1.24.0版本以下就可以解决这个问题。第三个问题就是训练集的yaml文件格式要参考coco的格式。首先第一个问题是在训练模型时会报错。
2021SC@SDUSC山东大学软件学院软件工程应用与实践——yolov5代码分析——第四篇——export.py(2)
qq_53219137的博客
12-27 1149
目录 parse_out函数 main run函数 请不要忽略代码中的注释 parse_out函数 设置opt参数 def parse_opt(): parser = argparse.ArgumentParser() parser.add_argument('--data', type=str, default=ROOT / 'data/coco128.yaml', help='dataset.yaml path') parser.add_argument('--w
YOLOv5 标签加载错误
Keeper的博客
10-17 404
转成：class x y h w格式。1、YOLOv5的标签格式错误。
解决libtorch在进行torchscript推理时程序崩溃或程序异常退出（Clion退出代码为 -1073740791 (0xC0000409)）
m0_61047553的博客
04-09 9097
在进行YOLOX的libtorch部署的时候，发现官方提供的代码里面解码的那一部分运行时间太长了，需要大概3秒钟，这个是远远不能达到应用需求的，但是作为一个土生土长的算法工程师，除了python以外，对其他的编程语言了解并不多，所以干脆就在python里面进行模型的解码，期望是模型输出的结果就是四个坐标点、分类以及置信度，但是在写完后处理之后，发现打包出来的torchscript在C++中不能forward（模型load是正常的），forward会导致程序异常退出，具体报错如下： 进程已结束，退出代码为
compute_point_cloud_distance(): incompatible function arguments. The following argument types are supported:
05-31
很抱歉，我的回答有误。`compute_point_cloud_distance()`方法已被弃用，无法使用。以下是一个新的示例代码，使用Open3D库将点云转换为深度图像： ``` python import open3d as o3d import numpy as np # 读取点云文件 pcd = o3d.io.read_point_cloud("point_cloud.pcd") # 获取点云的xyz坐标 xyz = np.asarray(pcd.points) # 创建相机参数 intrinsic = o3d.camera.PinholeCameraIntrinsic() intrinsic.set_intrinsics(640, 480, 525.0, 525.0, 319.5, 239.5) # 将点云投影到深度图像 depth = o3d.geometry.Image.create_from_depth_image( intrinsic, np.asarray(pcd.transform(intrinsic.intrinsic_matrix).depth)) # 将深度图像保存为图像文件 o3d.io.write_image("depth.png", depth) ``` 这段代码将读取名为"point_cloud.pcd"的点云文件，并将其转换为深度图像。转换后的深度图像将保存在名为"depth.png"的图像文件中。我们首先获取点云的xyz坐标，并创建了一个PinholeCameraIntrinsic对象，将图像的大小和相机内参设置为640x480，fx=525.0, fy=525.0, cx=319.5, cy=239.5。然后，我们使用`transform()`方法将点云转换为相机坐标系下的点云，并使用`create_from_depth_image()`方法将点云投影到深度图像。最后，我们将深度图像保存为图像文件。
评论 1
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值