当前位置: 首页 > news >正文

yolov5导出命令

python export.py --weights yolov5s.pt --img-size 640 --batch-size 1  --device cpu --include onnx  

关闭动态输入,cpu导出

检测onnx模型能否加载成功指令:
python detect.py --weights yolov5s.onnx --dnn   
终端调用detect.py检测图片命令:
python detect.py --weights yolov5s.onnx --img-size 640 --source .\data\images\bus.jpg  --device cpu
 
使用pt导出的yolov5s.onnx运行报错:

./main_cmake_exe
[ERROR:0@0.028] global onnx_importer.cpp:1061 handleNode DNN/ONNX: ERROR during processing node with 1 inputs and 1 outputs: [Floor]:(onnx_node!/model.11/Floor) from domain='ai.onnx'
terminate called after throwing an instance of 'cv::Exception'
  what():  OpenCV(4.8.0) /home/xwy/src/opencv-4.8.0/modules/dnn/src/onnx/onnx_importer.cpp:1083: error: (-2:Unspecified error) in function 'handleNode'
> Node [Floor@ai.onnx]:(onnx_node!/model.11/Floor) parse error: OpenCV(4.8.0) /home/xwy/src/opencv-4.8.0/modules/dnn/src/layers/elementwise_layers.cpp:260: error: (-215:Assertion failed) src.size == dst.size && src.type() == dst.type() && src.isContinuous() && dst.isContinuous() && src.type() == CV_32F in function 'forward'
>
已放弃 (核心已转储)

解决方法:导出onnx文件的时候把使用pytorch1.11或者更低版本

http://www.lryc.cn/news/500147.html

相关文章:

  • RabbitMQ的常用术语介绍
  • Docker魔法:用docker run -p轻松开通容器服务大门
  • 【后端面试总结】Redis过期删除策略
  • 数字图像处理(15):图像平移
  • 高级java每日一道面试题-2024年12月08日-JVM篇-什么是类加载器?
  • JAVA子类的无参构造器中第一行的super
  • mysql程序介绍,选项介绍(常用选项,指定选项的方式,特性),命令介绍(查看,部分命令),从sql文件执行sql语句的两种方法
  • Unity教程(十九)战斗系统 受击反馈
  • lanqiaoOJ 3744:小蓝的智慧拼图购物 ← pair+优先队列
  • Spring Boot教程之二十一:文件处理
  • 【Linux】Linux的基本常识+指令
  • Rocky Linux 9.3系统搭建Slurm环境【笔记】
  • 原生微信小程序使用原子化tailwindcss
  • 《掌握Nmap:全面解析网络扫描与安全检测的终极指南》
  • k8s-Informer概要解析(2)
  • UE5基本数据类型
  • Next.js 系统性教学:中间件与国际化功能深入剖析
  • 鸿蒙HarmonyOS元服务应用开发实战完全指导
  • CT中的2D、MPR、VR渲染、高级临床功能
  • 利用docker-compose来搭建flink集群
  • 力扣打卡10:K个一组翻转链表
  • 深度学习详解
  • 鸿蒙分享(一):添加模块,修改app名称图标
  • 【Redis】not support: redis
  • 【集群划分】含分布式光伏的配电网集群电压控制【33节点】
  • 嵌入式蓝桥杯学习5 定时中断实现按键
  • 【Java】类似王者荣耀游戏
  • C++<基本>:union是没有构造函数和析构函数的
  • SQL中IN和NOT操作符的用法
  • C++平常学习用的