mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmdeploy/demo
History
Chen Xin b85f34141b
[Feature] Support feature map output for mmsegmentation (#1625) 
* add feature map output for mmseg

* update api

* update demo

* fix return

* update format_shape

* fix lint

* update csharp demo

* update python demo && api

* fix coreml build

* fix lint

* better sort

* update

* update cpp demo & add missing header

* change to CHW

* update csharp api

* update isort version to 5.12.0

* fix python api

* fix log

* more detail api docs

* isort support python3.7

* remove isort change

* remove whitespace

* axes check

* remove FormatShapeImpl

* minor

* add permute tc

* remove stride buffer
 		2023-02-03 20:47:55 +08:00
..
csharp [Feature] Support feature map output for mmsegmentation (#1625) 2023-02-03 20:47:55 +08:00
csrc [Feature] Support feature map output for mmsegmentation (#1625) 2023-02-03 20:47:55 +08:00
java Add java apis and demos (#563) 2022-06-29 11:02:08 +08:00
python [Feature] Support feature map output for mmsegmentation (#1625) 2023-02-03 20:47:55 +08:00
resources add more images for demos and user guides (#1339) 2022-11-09 21:06:32 +08:00
tutorials 【Docs】Add ipython notebook tutorial (#234) 2022-04-27 20:04:56 +08:00
README.md [Docs] Replace markdownlint with mdformat and configure myst-parser (#610) 2022-06-17 09:19:10 +08:00
demo_rewrite.py
…

README.md

Demo

We provide a demo showing what our mmdeploy can do for general model deployment.

In demo_rewrite.py, a resnet18 model from torchvision is rewritten through mmdeploy tool. In our rewritten model, the forward function of resnet gets modified to only down sample the original input to 4x. Original onnx model of resnet18 and its rewritten are visualized through netron.

Prerequisite

Before we run demp_rewrite.py, we need to install pyppeteer through:

pip install pyppeteer

Demo results

The original resnet18 model and its modified one are visualized as follows. The left model is the original resnet18 while the right model is exported after rewritten.

Original resnet18 Rewritten model