[vision] support PaddleDetection/MaskRCNN model (#218)

* [vision] support padddetection maskrcnn

* [vision] fixed instance mask visualize func

* [vision] optimize instance mask visualize func

* [docs] update ppdet/maskrcnn docs

* [vision] update maskrcnn implementation

Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
DefTruth
2022-09-14 10:50:33 +08:00
committed by GitHub
parent 0dd9ecee65
commit 887b2d6ed6
22 changed files with 625 additions and 101 deletions

View File

@@ -39,6 +39,7 @@ fastdeploy.vision.detection.PaddleYOLOX(model_file, params_file, config_file, ru
fastdeploy.vision.detection.YOLOv3(model_file, params_file, config_file, runtime_option=None, model_format=Frontend.PADDLE)
fastdeploy.vision.detection.PPYOLO(model_file, params_file, config_file, runtime_option=None, model_format=Frontend.PADDLE)
fastdeploy.vision.detection.FasterRCNN(model_file, params_file, config_file, runtime_option=None, model_format=Frontend.PADDLE)
fastdeploy.vision.detection.MaskRCNN(model_file, params_file, config_file, runtime_option=None, model_format=Frontend.PADDLE)
```
PaddleDetection模型加载和初始化其中model_file params_file为导出的Paddle部署模型格式, config_file为PaddleDetection同时导出的部署配置yaml文件