Mask R-CNN-MXnet框架下的訓練與測試
阿新 • • 發佈:2019-01-26
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