MMDetection2系列-(3)-部署

MMDetection2系列-3-部署

@[TOC](MMDetection2系列-3-部署)
- 1. Pytorch to ONNX
- 2. ONNX to TensorRT

1. Pytorch to ONNX

pip install onnx-simplifier
pip install onnxruntime
python tools/deployment/pytorch2onnx.py configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py work_dirs/faster_rcnn_r50_fpn_1x_coco/epoch_10.pth --output-file work_dirs/faster_rcnn_r50_fpn_1x_coco/epoch_10.onnx --show --verify --simplify

2. ONNX to TensorRT

安装tensorrt
参考:deepstream 学习笔记

python tools/deployment/onnx2tensorrt.py work_dirs/mask_rcnn_r50_fpn_1x_coco/mask_rcnn_r50_fpn_1x_coco.onnx --trt-file work_dirs/mask_rcnn_r50_fpn_1x_coco/mask_rcnn_r50_fpn_1x_coco.trt --input-img demo/demo.jpg --shape 400 600 --mean 123.675 116.28 103.53 --std 58.395 57.12 57.375 --show --verify

参考:https://mmdetection.readthedocs.io/

MMDetection2系列-(3)-部署相关推荐

SCCM 2007 R2部署、操作详解系列之部署篇
SCCM 2007 R2部署.操作详解系列之部署篇导语: 大家好,我是John Cai(蔡建涛),欢迎大家来到我的博客-E网情深http://johncai.blog.51cto.com/.今天,我 ...
Jetson嵌入式系列模型部署-1
目录前言 1. What.Why and How 1.1 What 1.2 Why 1.3 How 2. tensorRT 2.1 什么是tensorRT? 2.2 tensorRT特性 2.3 t ...
小丸子学Hadoop系列之——部署Hbase集群
0.集群规划主机名 ip地址安装的软件运行的进程 AI-OPT-HBS01 10.46.52.30 hadoop,hbase namenode,zkfc,resourcemanager AI-O ...
Jetson嵌入式系列模型部署-3
目录注意事项一.2023/2/1更新前言 1. 源码下载 2. 环境配置 2.1 Jtop(option) 2.1.1 配置pip 2.1.2 安装jtop 2.1.3 使用jtop 2.2 编 ...
Jetson嵌入式系列模型部署-2
目录前言 1. 源码下载 2. 环境配置 2.1 Jtop(option) 2.1.1 配置pip 2.1.2 安装jtop 2.1.3 使用jtop 2.2 源码配置说明 3. 运行 3.1 .p ...
Elasticsearch系列-安装部署
Elasticsearch系列-安装部署准备安装错误准备 1.服务器创建es用户 useradd es passwd es 2.elasticsearch部署环境JDK 3.安装包可以去el ...
Flask 系列之部署发布
说明操作系统:Windows 10 Python 版本:3.7x 虚拟环境管理器:virtualenv 代码编辑器:VS Code 实验目标通过 Windows 的 WSL,将我们的项目网站部署到 ...
利用福禄克DSX系列测试仪部署MPTL模块化插头端接链路
在与客户合作时,认识到MPTL(模块化插头端接链路)比使用传统插座.面板和设备跳线更适合连接设备的地方非常重要.如果您要部署这种,福禄克网络指定经销商-明辰智航的工程师会在测试时介绍给您. 如果您与零 ...
运维与自动化系列③自动化部署基础与shell脚本实现
自动化部署基础与shell脚本实现关于自动化的基础知识: 1.1:当前代码部署的实现方式: 运维纯手工scp到web服务器纯手工登录git服务器执行git pull或svn服务器执行svn upd ...

MMDetection2系列-(3)-部署

MMDetection2系列-3-部署

1. Pytorch to ONNX

2. ONNX to TensorRT

MMDetection2系列-(3)-部署相关推荐

最新文章

热门文章