mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity
mmdeploy/docs/en/04-supported-codebases/mmedit.md
Junhwa Song f7c484a046
Add support for converting a inpainting model to ONNX and TensorRT (#1831) 
* Add support for inpainting models

* Add configs

* Add comment

* Refactor

* Add test code for inpainting task

* Fix

* Fix

* Update

* Fix

* Fix

* Update docs

* Update

* Fix visualization

* Handle case without Resize
2023-03-29 19:17:24 +08:00

3.3 KiB
Raw Blame History

MMEditing Support

MMEditing is an open-source image and video editing toolbox based on PyTorch. It is a part of the OpenMMLab project.

MMEditing installation tutorial

Please refer to official installation guide to install the codebase.

MMEditing models support

Model Task ONNX Runtime TensorRT ncnn PPLNN OpenVINO Model config
SRCNN super-resolution Y Y Y Y Y config
ESRGAN super-resolution Y Y Y Y Y config
ESRGAN-PSNR super-resolution Y Y Y Y Y config
SRGAN super-resolution Y Y Y Y Y config
SRResNet super-resolution Y Y Y Y Y config
Real-ESRGAN super-resolution Y Y Y Y Y config
EDSR super-resolution Y Y Y N Y config
RDN super-resolution Y Y Y Y Y config
Global&Local* inpainting Y Y N N N config
DeepFillv1* inpainting Y Y N N N config
PConv* inpainting Y Y N N N config
DeepFillv2* inpainting Y Y N N N config
AOT-GAN* inpainting Y Y N N N config
  1. We skipped quantitative evaluation for image inpainting due to the high computational cost required for testing.