Deepstream yolov5 两种引擎(engine)生成方式

方式一、通过cfg和wts文件在deepstream中直接生成trt引擎

DeepStream-Yolo工程

./DeepStream-Yolo/utils/gen_wts_yoloV5.py

执行如下命令可生成yolov5s.cfg和yolov5s.wts两个文件：

python3 gen_wts_yoloV5.py -w yolov5s.pt

将以上生成的两个文件yolov5s.cfg和yolov5s.wts复制到/opt/nvidia/deepstream/deepstream-6.0/sources/deepstream_python_apps/apps/deepstream-yolov5文件夹下，该文件夹下config_infer_primary.txt文件内容改成如下所示：

[property]
gpu-id=0
net-scale-factor=0.0039215697906911373
model-color-format=0
custom-network-config=yolov5s.cfg
model-file=yolov5s.wts
model-engine-file=model_b1_gpu0_fp32.engine
labelfile-path=labels.txt
batch-size=1
network-mode=0
num-detected-classes=80
interval=0
gie-unique-id=1
process-mode=1
network-type=0
cluster-mode=4
maintain-aspect-ratio=1
parse-bbox-func-name=NvDsInferParseYolo
custom-lib-path=nvdsinfer_custom_impl_Yolo/libnvdsinfer_custom_impl_Yolo.so
engine-create-func-name=NvDsInferYoloCudaEngineGet

3.运行视频文件检测程序并同时生成trt引擎文件(需要几分钟时间)：
python3 deepstream_yolov5_file.py file:///opt/nvidia/deepstream/deepstream-6.0/samples/streams/sample_qHD.h264

方式二、通过tensorrtx项目生成trt文件，具体实现方式可参照项目内README.md文件

Deepstream yolov5 两种引擎(engine)生成方式相关推荐

数据库的两种引擎Innodb和MyIASM
Innodb引擎 Innodb引擎提供了对数据库ACID事务的支持,并且实现了SQL标准的四种隔离级别,关于数据库事务与其隔离级别的内容请见数据库事务与其隔离级别这篇文章.该引擎还提供了行级锁和外键 ...
mysql常用的两种数据库引擎_Mysql数据库常用两种引擎的介绍
1.数据库引擎介绍 MySQL数据库引擎取决于MySQL在安装的时候是如何被编译的.要添加一个新的引擎,就必须重新编译MYSQL.在MySQL数据库中,常用的引擎主要就是2个:Innodb和MyIAS ...
mysql两种引擎的适用场景_MySQL两种引擎的区别和应用场景
Innodb引擎 Innodb引擎提供了对数据库ACID事务的支持,并且实现了SQL标准的四种隔离级别.该引擎还提供了行级锁和外键约束,它的设计目标是处理大容量数据库系统,它本身其实就是基于MySQL ...
mysql的两种存储方法_mysql的两种存储引擎
背景最近在使用spring-data-jpa操作mysql,在使用jpa的自动建表功能之后,处理多对多关系@ManyToMany的时候,spring-data-jpa会给中间表的两个字段都加上外键引 ...
mysql 两种存储引擎 MyISAM 和InnoDB
mysql 两种存储引擎 MyISAM 和InnoDB MyISAM 是MySQL中默认的存储引擎,一般来说不是有太多人关心这个东西.决定使用什么样的存储引擎是一个很tricky的事情,但是还是值我们 ...
MySQL 两种存储引擎：MyISAM与InnoDB对比及理解
MySQL 两种存储引擎:MyISAM与InnoDB对比及理解目前MySQL默认的存储引擎是InnoDB 现在大多数时候我们使用的都是InnoDB存储引擎,但是在某些情况下使用MyISAM更好,比如 ...
mysql中两种备份方法的优缺点_Mysql两种存储引擎的优缺点
MyISAM引擎是一种非事务性的引擎,提供高速存储和检索,以及全文搜索能力,适合数据仓库等查询频繁的应用.MyISAM中,一个table实际保存为三个文件,.frm存储表定义,.MYD存储数据,.MY ...
mysql 有两种数据库引擎发音
mysql 有两种数据库引擎一种是 MyISAM,一种是 InnoDB MyISAM 发音为 "my-z[ei]m"; InnoDB 发音为 "in-no-db&quo ...
mysql存储引擎的区别_Mysql的两种存储引擎以及区别
一.Mysql的两种存储引擎 1.MyISAM: ①不支持事务,但是整个操作是原子性的(事务具备四种特性:原子性.一致性.隔离性.持久性) ②不支持外键,支持表锁,每次所住的是整张表 MyISAM的表 ...

Deepstream yolov5 两种引擎(engine)生成方式

方式一、通过cfg和wts文件在deepstream中直接生成trt引擎

方式二、通过tensorrtx项目生成trt文件，具体实现方式可参照项目内README.md文件

Deepstream yolov5 两种引擎(engine)生成方式相关推荐

最新文章

热门文章