PaddleX目前提供了MaskRCNN实例分割模型结构,多种backbone模型,可满足开发者不同场景和性能的需求。
| 模型(点击获取代码) | Box MMAP/Seg MMAP | 模型大小 | GPU预测速度 | Arm预测速度 | 备注 |
|---|---|---|---|---|---|
| MaskRCNN-ResNet50-FPN | 38.7%/34.7% | 170.0MB | 160.185ms | - | 模型精度高,适用于服务端部署 |
| MaskRCNN-ResNet18-FPN | -/- | 120.0MB | - | - | 模型精度高,适用于服务端部署 |
| MaskRCNN-HRNet-FPN | 38.7%/34.7% | 116.MB | - | - | 模型精度高,预测速度快,适用于服务端部署 |
代码保存到本地后,即可直接训练,训练代码会自动下载训练数据开始训练
如保存为
mask_r50_fpn.py,如下命令即可开始训练python mask_r50_fpn.py