Mask R-CNN-MXnet框架下的训练与测试

因为之前对Faster R-CNN编译训练过，所以同样使用mxnet框架下的源码：https://github.com/TuSimple/mx-maskrcnn

mxnet的安装见点击打开链接

Faster R-CNN的编译训练见点击打开链接

Mxnet一定要加入ROIAlign重新编译！！！

由于我安装mxnet时是按照Faster R-CNN的需求安装的，但Mask R-CNN用到了ROIAlign，需要将将rcnn/CXX_OP/下的文件夹复制到了mxnet/src/operator/中，重新编译，否则会在训练到一半的时候报错AttributeError: 'module' object has no attribute 'ROIAlign'（make不会报错，第8次迭代时报错）。

所以一定要加入ROIAlign编译，make是发现不了错误的TT

另外，有些头文件路径需要修改。

1 训练

具体步骤参考：点击打开链接

测试集：Cityscapes数据集

共训练10类：

"bicycle" "caravan" "motorcycle""rider" "bus" "train" "car" "person""truck" "trailer"

make

bash scripts/train_alternate.sh

用4个gpu训练，速度仍然比Faster R-CNN慢很多。

2 测试

使用训练好的模型对测试集中的图片进行测试

bash scripts/download_cityscapescripts.sh

bash scripts/eval.sh

bash scripts/demo_single_image.sh