IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 游戏开发 -> 转权重格式darknet转onnx转tensorrt -> 正文阅读

[游戏开发]转权重格式darknet转onnx转tensorrt

将darknet格式的权重转为onnx格式,然后再转为tensorrt格式
首先准备好一个darknet格式的权重文件(.weights),配置文件(.cfg),训练集名称文件(.names)

一、darknet转为onnx

这里我使用的是这个开源链接
https://github.com/Tianxiaomo/pytorch-YOLOv4
在这里插入图片描述

图1

通过一个python脚本,将darknet格式的权重,转为onnx格式,并且可以输入一张图片,这个脚本会对这张图片进行识别 具体的步骤为: (1)先把上面这个github里面的文件下载下来 (2)运行命令:
python demo_darknet2onnx.py yolov4-tiny-416.cfg obj.names yolov4-tiny-416.weights fire1.jpg 1

按照上面图1的格式来写这个命令就行
运行完成之后,就会生成一个.onnx格式的权重了。
在这里插入图片描述

二、将onnx格式转为tensorrt格式

这里也是参考上面github里面的
在这里插入图片描述

trtexec --onnx=<onnx_file> --explicitBatch --saveEngine=<tensorRT_engine_file> --workspace=<size_in_megabytes> --fp16

tensorrt格式的权重好像有两种,一种是.trt,一种是.engine。我这里转化的是.trt格式,.engine格式没试过。
运行命令:

/opt/TensorRT-8_0_1_6/bin/trtexec --onnx=yolov4_1_3_416_416_static.onnx --explicitBatch --saveEngine=yolov4-tiny-416.trt --workspace=2048 --fp16

(这个转格式时间有点长,可能得个10分钟)
备注:
如果提示没有trtexec,需要安装tensorrt,然后在tensorrt的安装文件夹下面的samples里面

cd <TensorRT root directory>/samples/trtexec
make

编译之后,在<TensorRT root directory>/bin里面就有trtexec了。


参考链接:
https://github.com/Tianxiaomo/pytorch-YOLOv4

  游戏开发 最新文章
6、英飞凌-AURIX-TC3XX: PWM实验之使用 GT
泛型自动装箱
CubeMax添加Rtthread操作系统 组件STM32F10
python多线程编程:如何优雅地关闭线程
数据类型隐式转换导致的阻塞
WebAPi实现多文件上传,并附带参数
from origin ‘null‘ has been blocked by
UE4 蓝图调用C++函数(附带项目工程)
Unity学习笔记(一)结构体的简单理解与应用
【Memory As a Programming Concept in C a
上一篇文章      下一篇文章      查看所有文章
加:2022-03-22 20:55:47  更:2022-03-22 20:57:23 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2025年1日历 -2025/1/16 17:54:30-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码