日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

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

欄目分類
最近更新