Mask_RCNN模型自训练数据,搭建指南(win10,pycharm,TensorFlow)
1.安装Anconda环境(请自行百度安装)
2.创建一个Anconda虚拟环境
conda create -n Tensorflow_envs python=3.6
3.打开PyCharm
4.源代码下载
git clone https://github.com/matterport/Mask_RCNN.git
5.安装环境依赖
pip install –r requirements.txt
如果安装不成功的话,请加上豆瓣源或者其他源
pip install –r requirements.txt -i https://pypi.douban.com/simple
6.下载预训练模型
https://github.com/matterport/Mask_RCNN/releases
7.安装pycocotools
pip install pycocotools -i https://pypi.douban.com/simple
8.安装jupyter
pip install jupyter -i https://pypi.douban.com/simple
9.运行Mask_RCNN项目下的/samples/demo.ipynb,如果能够运行说明环境基本上搭建完成了,基本上到这一步都没有什么问题。
关键在于训练自己的数据集(关键点)
1