maskrcnn-benchmark 工程配置

SpikeKing發表於2019-05-08

maskrcnn-benchmark是Facebook開源的基準(benchmark)演算法,其中包含檢測、分割和人體關鍵點檢測。

工程:github.com/facebookres…

Faster R-CNN and Mask R-CNN in PyTorch 1.0

測試圖片

  1. 如何使用pip環境配置工程;
  2. 如何使用MacOS的CPU環境配置依賴庫;

環境:MacOS + pip + torch + maskrcnn-benchmark


基礎配置

下載maskrcnn-benchmark工程

git clone https://github.com/facebookresearch/maskrcnn-benchmark.git
複製程式碼

virtualenv建立虛擬環境,選擇python3,同時啟用。

virtualenv -p python3 mlp3_venv
複製程式碼

必須使用Python3.6.8+,推薦使用3.7

或者 使用已有的虛擬環境。

安裝依賴包:

pip install ninja yacs cython matplotlib tqdm opencv-python
複製程式碼

安裝PyTorch,直接官網

pip3 install torch torchvision
複製程式碼

Mac + PyTorch

依賴庫

maskrcnn-benchmark需要兩個依賴庫cocoapi和apex,參考略有不同

安裝包:pycocotools 2.0,啟用已有的虛擬環境,再執行:

git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
python setup.py build_ext install
複製程式碼

安裝包:apex 0.1,執行:

git clone https://github.com/NVIDIA/apex.git
cd apex
MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install --cpp_ext
複製程式碼

注意:

  • MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++是Mac所特有,指定編譯的C++庫;
  • CPU環境,不新增--cuda_ext

安裝包:maskrcnn-benchmark 0.1,執行

cd maskrcnn-benchmark
MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py build develop
複製程式碼

測試

·.torch/models中,下載模型,約458M:

wget https://dl.fbaipublicfiles.com/detectron/37697547/12_2017_baselines/e2e_keypoint_rcnn_R-50-FPN_1x.yaml.08_42_54.kdzV35ao/output/train/keypoints_coco_2014_train%3Akeypoints_coco_2014_valminusminival/generalized_rcnn/model_final.pkl
複製程式碼

重新命名:

mv model_final.pkl _detectron_37697547_12_2017_baselines_e2e_keypoint_rcnn_R-50-FPN_1x.yaml.08_42_54.kdzV35ao_output_train_keypoints_coco_2014_train%3Akeypoints_coco_2014_valminusminival_generalized_rcnn_model_final.pkl
複製程式碼

預測


Troubleshooting

常見問題的解決方案。

libomp.dylib

遇到如下問題:libomp.dylib無法載入

ImportError: dlopen(python3.7/site-packages/torch/_C.cpython-37m-darwin.so, 9): Library not loaded: /usr/local/opt/libomp/lib/libomp.dylib
  Referenced from: python3.7/site-packages/torch/lib/libshm.dylib
  Reason: image not found
複製程式碼

安裝libomp包:

brew install libomp
複製程式碼

參考:github.com/pytorch/pyt…

torch.version.cuda.split('.')

GPU的版本,異常:

get_cuda_version
    return tuple(int(x) for x in torch.version.cuda.split('.'))
AttributeError: 'NoneType' object has no attribute 'split'
複製程式碼

環境是Mac的CPU環境,沒有GPU,而在apex原始碼中,依然尋找GPU,導致Bug,修改原始碼即可:

/apex/amp/lists/torch_overrides.py的第69行,增加Try-Except語句,遮蔽異常:

try:
    if utils.get_cuda_version() >= (9, 1, 0):
        FP16_FUNCS.extend(_bmms)
    else:
        FP32_FUNCS.extend(_bmms)
except:
    FP32_FUNCS.extend(_bmms)
複製程式碼

OK, that's all!

相關文章