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

暑期编程PK赛 得CSDN机械键盘等精美礼品！