1. 程式人生 > >Mask R-CNN-MXnet框架下的訓練與測試

Mask R-CNN-MXnet框架下的訓練與測試

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