yolov5 trt部署
时间: 2023-09-05 20:13:12 浏览: 58
要将Yolov5部署到TensorRT中,需要执行以下步骤:
1. 将`gen_wts.py`文件复制到Yolov5的根目录下,使用命令`cp tensorrtx/yolov5/gen_wts.py yolov5` [1。
2. 进入Yolov5的根目录,使用命令`cd yolov5` [2。
3. 激活之前创建的conda虚拟环境,使用命令`conda activate yolov5` [2。
4. 生成`yolov5s.wts`文件,使用命令`python gen_wts.py -w yolov5s.pt -o yolov5s.wts` [2。
5. 进入工作目录,使用命令`cd myworkspace` [3。
6. 将`yolov5s.wts`文件复制到TensorRT的Yolov5路径下,使用命令`cp yolov5/yolov5s.wts tensorrtx/yolov5` [3。
7. 进入TensorRT的Yolov5路径,使用命令`cd tensorrtx/yolov5` [3。
现在,你已经完成了Yolov5的TensorRT部署。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [triton部署yolov5笔记(三)](https://blog.csdn.net/weixin_41796280/article/details/126160238)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]