可可数据集上的预训练模型正在进行中。 分支可可还不稳定。 由于可可分行不够稳定,因此目前仅用作教育目的:
TL; DR:
演示在demo.ipynb
Doc在doc/mask-rcnn-intro.ipynb
配置-> configuration.py
test使用那些网络标题
面罩RCNN PyTorch
Mask-RCNN的PyTorch 0.4实现。 这是我们在工作的副项目
特征:
与设备无关的代码。 在不修改代码的情况下同时运行gpu / cpu,培训和测试都不需要gpu。 感谢pytorch 0.4!
完整记录的代码,带有jupyter笔记本指南,易于使用的配置
清晰的代码结构,带有完整的单元测试,以最小的痛苦扩展
要求
代码在以下系统下测试
Ubuntu 16.04(CPU / Nvidia GPU)
macOS High Sierra(CPU版本)
Win
1