| 模型 | Top1精度 | 模型大小 | GPU预测速度 | Arm预测速度 | 备注 | | :---------------- | :------- | :------- | :--------- | :--------- | | | MaskRCNN-ResNet50-FPN | 97.5% | 22M | 10ms | 200ms | | | MaskRCNN-ResNet101-FPN | | | | | | | MaskRCNN-HRNet-FPN | | | | | |
将对应模型的训练代码保存到本地后,即可直接训练,训练代码会自动下载训练数据开始训练,如保存为mask_r50_fpn.py,如下命令即可开始训练
python mask_r50_fpn.py