mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
* change docs from 1.x to main * bump version * update readme status info and highlights * revert version change * fix dead links * fix script * fix docs * fix building status * fix md lint error * upgrade to ubuntu20.04 for github ci * Revert "upgrade to ubuntu20.04 for github ci" This reverts commit 9c598f4857633dfb96b116ca9fb9a5d15b5fa6b4. * update codebase md files * upgrade to ubuntu20.04 for github ci * fix * fix requirements * fix * fix ci * bump version * fix * fix pplnn torchscript * fix building status * fix ncnn * fix * remove circileci hold * fix pytorch image
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 |
---|---|
![]() |
![]() |