facebook -------maskrcnn-benchmark

1、安装步骤

参考网站:https://blog.csdn.net/mango_haoming/article/details/84257401?tdsourcetag=s_pcqq_aiomsg

安装:由xx安装

安装maskrcnn_benchmark

在cuda安装完成,gcc也成功降级之后,到这一步直接按照maskrcnn_benchmark的官方教程按照就可以了,只不过要安装opencv,就稍加修改。中文部分就是添加的部分

# first, make sure that your conda is setup properly with the right environment
# for that, check that `which conda`, `which pip` and `which python` points to the
# right path. From a clean conda env, this is what you need to do

conda create --name maskrcnn_benchmark
source activate maskrcnn_benchmark

#安装opencv
conda install -c conda-forge opencv
#安装opencv

# this installs the right pip and dependencies for the fresh python
conda install ipython

# maskrnn_benchmark and coco api dependencies
pip install ninja yacs cython matplotlib

# follow PyTorch installation in https://pytorch.org/get-started/locally/
# we give the instructions for CUDA 9.0
conda install pytorch-nightly -c pytorch

# install torchvision
cd ~/github
git clone https://github.com/pytorch/vision.git
cd vision
python setup.py install

# install pycocotools
cd ~/github
git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
python setup.py build_ext install

# install PyTorch Detection
cd ~/github
git clone https://github.com/facebookresearch/maskrcnn-benchmark.git
cd maskrcnn-benchmark
# the following will install the lib with
# symbolic links, so that you can modify
# the files if you want and won't need to
# re-build it
python setup.py build develop

# or if you are on macOS
# MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py build develop

#到这里发现opencv失效,但是conda list查看包还在,无奈只能重新安装,但命令变化了
pip install opencv-python

2、测试

3、训练:

#进入maskrcnn-benchmark目录下,激活maskrcnn_benchmark虚拟环境
[@localhost]$ cd maskrcnn-benchmark
[@maskrcnn-benchmark]$ source activate maskrcnn_benchmark
#指定模型配置文件,执行训练启动脚本
(maskrcnn_benchmark) handkoo@handkoo$python tools/train_net.py --config-file configs/e2e_mask_rcnn_R_101_FPN_1x.yaml

类似结果:

4、验证:

#指定模型配置文件,执行测试启动脚本
(maskrcnn_benchmark) handkoo@handkoo$python tools/test_net.py --config-file configs/e2e_mask_rcnn_R_101_FPN_1x.yaml

5、测试

(maskrcnn_benchmark) handkoo@handkoo:~/disk/github/maskrcnn-benchmark$ python tools/inference.py

猜你喜欢

转载自blog.csdn.net/liulina603/article/details/84968904