C++推理支持mask-rcnn吗?
已关闭
C++推理支持mask-rcnn吗?
Created by: cvfzhong
试了下用faster-rcnn的配置好像不行 请问需要做什么样的修改才能适配mask-rcnn?
Created by: qingqing01
@cvfzhong 注意,使用C++推理时,需要先使用tools/export_model.py导出模型,再使用C++推理程序。 MaskRCNN的输入需要3个输入image、image_info(resize的信息)、image_shape(原始图像大小),参考: https://github.com/PaddlePaddle/PaddleDetection/blob/release/0.2/docs/advanced_tutorials/inference/EXPORT_MODEL.md#%E6%A8%A1%E5%9E%8B%E5%AF%BC%E5%87%BA
C++预测的配置,可以参考RCNN-FPN: https://github.com/PaddlePaddle/PaddleDetection/blob/release/0.2/inference/conf/detection_rcnn_fpn.yaml
另外,您的部署如果不是一定必需依赖C++的话,可以参考Python的预测代码。 相比C++还是要简单很多。同时我们也在重构C++预测,使得代码更容易读懂、以及使用更简单。
Created by: qingqing01
@cvfzhong
https://github.com/PaddlePaddle/PaddleDetection/blob/master/tools/cpp_infer.py
master分支上,有合入新的,这个文件完全脱离PaddleDetection可以用,也是适配了MaskRCNN。您用下这个呢