mask rcnn mxnet

这篇博客详细介绍了如何在Ubuntu环境下,基于MXNet框架实现Mask R-CNN。首先,从GitHub克隆源代码,避免下载特定版本的MXNet。接着,下载Cityscapes数据集、Resnet-50预训练模型,并构建ROIAlign运算符。然后,编译MXNet源代码,安装Python包,训练模型。博主遇到了训练过程中的错误并分享了解决方案,最终完成了训练并进行了模型评估。
摘要由CSDN通过智能技术生成

GitHub地址:https://github.com/TuSimple/mx-maskrcnn

首先表达一下对凯明大神和RBG大神的膜拜!

20171124  一定要在终端上使用命令行下载源代码,因为mxnet官网上会建议下载0.11.0版本,但是这个版本有问题,在后续训练中会报错,所以直接clone,这样会下载最新的mxnet

git clone --recursive https://github.com/TuSimple/mx-maskrcnn.git

训练准备

1.首先下载Cityscapes数据(gtFine_trainvaltest.zip和leftImg8bit_trainvaltest.zip)下载地址是https://www.cityscapes-dataset.com/,它需要注册,并等待管理员进行确认,大家可以先试着下载,下载速度还可以。


2.下载Resnet-50预训练模型

bash scripts/download_res50.sh         得到resnet-50-0000.params

3.使用ROIAlign运算符构建MXNet

cp rcnn/CXX_OP/* incubator-mxnet/src/operator/

将rcnn/CXX_OP/下的文件夹复制到了 incubator-mxnet/src/operator/

(我一开始是先安装incubator-mxnet,后拷贝进去,一直有错,一定要先拷贝进去,将align也编译进去,不然一直会报AttribuuteError: 'module' object has no attribute 'ROIAlign'


从源代码建立MXNet请参考https://mxnet.incubator.apache.org/get_started/build_from_source.html(如果是从GitHub上下载的源代码,则目录中有incubator-mxnet,这边就不需要下载了)

我是在Ubuntu14.04下安装mxnet

3.1首先需要Ubuntu>=13.10,debian >=8      

sudo apt-get update && sudo apt-get install build-essential git

3.2需要BLAS库,可以安装ATLAS、OpenBLAS、MKL,我安装的是atlas

sudo apt-get install libatlas-base-dev

3.3需要opencv库

sudo apt-get install libopencv-dev

3.4需要cuda和cudnn用于加速,cuda7.5和8.0都需要cudnn5

wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_7
  • 4
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 39
    评论
评论 39
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值