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

PyTorch入门实战教程

今天,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入门实战教程
除特别注明外,本站所有文章均为 PyTorch 中文网原创,转载请注明出处:https://www.pytorchtutorial.com/pytorch-1-0-faster-r-cnn-mask-r-cnn/

Leave a Reply

Your email address will not be published. Required fields are marked *

返回顶部