0°

官方 PyTorch 1.0 实现 Faster R-CNN 和 Mask R-CNN

今天,Facebook Research 团队在 Github 上更新了用 PyTorch 1.0 实现的 Faster R-CNN 和 Mask R-CNN,为了让大家可以用 PyTorch 1.0 更加方便地创建图像识别和 segmentation 相关的项目。

alt text

项目特点

  • PyTorch 1.0: RPN, Faster R-CNN 和 Mask R-CNN 的实现和我们的 Detectron 实现的准确率相当甚至更好
  • 速度很快: 比 Detectron 快 2 倍,比 mmdetection 训练速度快 30%
  • 高效的内存管理: 在训练阶段比 mmdetection 少占用 500MB 的 GPU 内存
  • 支持多 GPU 训练和推理
  • 分批推理: 支持一次输入多张图片
  • 支持使用 CPU 推理
  • 提供几乎所有的 Mask R-CNN 和 Faster R-CNN 1x schedule 配置的预训练模型

安装需求

  • PyTorch 1.0 (nightly release)
  • torchvision
  • cocoapi
  • yacs
  • matplotlib
  • (可选) OpenCV for the webcam demo

安装命令实例如下:

Github 项目地址

本项目的 Github 地址:https://github.com/facebookresearch/maskrcnn-benchmark

PyTorch入门实战教程
2 条回复 A 作者 M 管理员
  1. 跑不起来呀, 大家能用吗

  2. 请问一定需要CUDA运行吗?虚拟机下如何跑demo呢?

欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论