maskrcnn_benchmark
maskrcnn-benchmark的使用
项目地址和安装步骤
facebook官方推出的mask rcnn的pytorch实现。与其他版本相比,有更快训练速度和更高的识别速度,支持多GPU训练,且完全基于pytorch实现。
https://github.com/facebookresearch/maskrcnn-benchmark/blob/master/INSTALL.md
按照给出的安装步骤即可使用。
数据集
只在coco数据集下成功测试。VOC数据集在下载之后,用于训练时老是提示缺少相关文件,好像缺的文件在某宝上可以买到,没有尝试可不可用。
coco数据集下载连接
http://images.cocodataset.org/zips/train2014.zip
http://images.cocodataset.org/annotations/annotations_trai