TritonObjDet新增要素的方法

一、确定新增要素的名字

要素名字不统一，会造成后续环境出错，这里以地面文字为例子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

注意

编译3090和2080的时候，有个地方需要修改，否则编译会报错，3090的时候加上这一行，2080用镜像triton_lhh2,3090用镜像3090

二、标牌新增类型

检查了好多地方，都没有发现代码类别还有哪里没有修改，这是就要检查一下模型本身。有时候模型弄混了，模型是错的。