Open WuChannn opened 1 year ago
@AllentDan will you plan to support MaskRCNN?
We recommend you use other algorithms like DBNet or PSENet. They are better than MRCNN. If you still want to use MRCNN to do text detection. You may try mmocr 1.x and mmdeploy 1.x. MMOCR 1.x directly uses MMDet models instead of maintaining codes of MRCNN in 0.x itself.
BTW, we welcome your PR to MMDeploy.
@AllentDan Are you sure that mmdeploy 1.x can convert mmocr 1.x model to onnx? I have spent a lot of time converting.
@AllentDan Are you sure that mmdeploy 1.x can convert mmocr 1.x model to onnx? I have spent a lot of time converting.
Yes, whats's the error?
Hi, @WuChannn . MRCNN of MMOCR is now supported, you may try PR #1534 .
Checklist
Describe the bug
mmdeploy/codebase/mmocr/deploy/text_detection.py 298行self.model_cfg.model.bbox_head,然而mmocr中的maskrcnn的模型config中的bbox_head为model.roi_head.bbox_head,因此这里应该是有个bug
Reproduction
直接使用mmdeploy将mmocr训练得到的torch模型转化为onnx模型时就会出现
Environment
Error traceback
No response