学无先后,达者为师

网站首页 编程语言 正文

TritonObjDet新增要素的方法

作者:上天夭 更新时间: 2022-07-16 编程语言

一、确定新增要素的名字

要素名字不统一,会造成后续环境出错,这里以地面文字为例子grountext

二、plan_build_script新增

在这里插入图片描述

1、weights转onnx模型

(1)demo_darknet2onnx.py

首先根绝,model.names确定类别,修改num_classes,有两处

(2)model.cfg

确认width,height是否都等于896

(3)开始转

cd convert_onnx_script
python demo_darknet2onnx.py model.cfg model.weights sample_img.jpg 1
cd ..
mv convert_onnx_script/model_1_3_896_896.onnx ./model.onnx

如果遇到没有安装的工具包,直接pip install

2、onnx转trt模型

(1)修改num_class

(2)修改setName

warningarea_bbox_out,这里的名字一定要和其他地方的保持一致

(3)编译执行

make -j4
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./plugin
./yolov4_builder

三、model_new新增

在这里插入图片描述

1、修改labels_cls.txt,labels_had.txt

2、替换model.plan

3、修改config.pbtxt

name有两处需要修改
在这里插入图片描述

4、修改detection_ensemble

需要添加两个部分,一个是input部分,一个是output部分,注意名字前后都要保持一致
在这里插入图片描述
在这里插入图片描述

四、triton_universe修改

1、GlobalHub.cc

在这里插入图片描述

2、WriteRedis.cc

在这里插入图片描述

原文链接:https://blog.csdn.net/weixin_38952721/article/details/125778834

栏目分类
最近更新