https://github.com/matterport/Mask_RCNN
首先放上github
坑贼多!!
一.配置conda环境
conda create -n maskrcnn python=3.6
conda activate maskrcnn
二.安装tensorflow-gpu 1.9.0
(不要问为什么,踩过1.14.0的坑,好多api改了,一运行一堆bug!一把辛酸泪)
如果不想自己搞cudnn和cuda的话,一键安装支持gpu的最低版本是tensorflow-gpu 1.9.0 所以安装这个就可以了
conda install tensorflow-gpu=1.9.0
安装完以后,按照github上安装各种依赖!
conda install scikit-image
conda install imgaug
有两个要特别注意的:
(1) keras
conda install keras=2.1.3
tensorflow1.9.0必须配keras=2.1.3 如果不带的话,他默认会给你装上2.3.1,但是这个和tensorflow1.9.0是不匹配的。这也是个坑。
(2) pycocotools
git clone https://github.com/cocodataset/cocoapi.git
cd cocoapi/PythonAPI
conda activate maskrcnn
make
make install
记住,在make install前一定要激活tensorflow所在的环境