MaskRCNN学习(一)——win10环境下配置运行

1. 项目地址

https://github.com/matterport/Mask_RCNN

2. 安装过程

2.1 创建虚拟环境

打开Anaconda Prompt进入保存的项目地址,执行conda create -n maskrcnn python=3.6命令。

成功创建环境后,执行conda activate maskrcnn命令,进入环境。

2.3 安装Tensorflow和Keras

注意 Tensorflow和Keras具有对应关系,可参考https://www.cnblogs.com/carle-09/p/11661261.html
在这里我使用的是tensorflow 1.14.0 + Keras 2.2.5

运行命令

pip install tensorflow-gpu==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

没有GPU就用

pip install tensorflow==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

安装Keras

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple keras==2.2.5
2.3 根据requirements.txt安装相应的依赖

运行命令pip install -r requirements.txt

注意 如果用上一条命令安装的太慢,可以使用国内镜像逐个安装,例如安装matplotlib:

pip install matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple

其他的包改一下名字就可以。

输入conda list查看已安装的包,看看是否安装成功

3. 下载训练好的COCO权重mask_rcnn_coco.h5

直接在github下载太慢,建议使用网盘进行下载
链接:https://pan.baidu.com/s/1wzfsBREJHtYORBo8oC1P_g
提取码:gsud

4. 安装pycocotools

4.1 安装vc++ 2015构建工具

https://download.microsoft.com/download/5/f/7/5f7acaeb-8363-451f-9425-68a90f98b238/visualcppbuildtools_full.exe
下载后安装

4.2 安装pycocotools

输入命令pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI

5. 使用jupyter notebook运行demo.ipynb

直接在命令行中输入jupyter notebook,显示如下
在这里插入图片描述
打开sample目录下的demo.ipynb文件并运行。

最终运行成功的结果如下图所示:

在这里插入图片描述

  • 0
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
Windows下配置 maskrcnn-benchmark 环境需要遵循以下步骤: 1. 首先,确保计算机已安装好 Python3 和 CUDA。如果没有,请先安装这些依赖项。 2. 在命令行中使用 Git 命令克隆 maskrcnn-benchmark 仓库。可以使用以下命令: ``` git clone https://github.com/facebookresearch/maskrcnn-benchmark.git ``` 3. 进入克隆的仓库目录: ``` cd maskrcnn-benchmark ``` 4. 使用以下命令创建并激活 Python 虚拟环境: ``` python -m venv .env .env\Scripts\activate ``` 5. 安装 PyTorch。可以从 PyTorch 官方网站选择适合你的 CUDA 和操作系统的版本进行安装。例如,如果你的 CUDA 版本是 10.1,可以使用以下命令安装: ``` pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 -f https://download.pytorch.org/whl/cu101/torch_stable.html ``` 6. 安装依赖项。运行以下命令来安装所需的依赖项: ``` pip install -r requirements.txt ``` 7. 编译 maskrcnn_benchmark。运行以下命令完成编译: ``` python setup.py build develop ``` 8. 配置 COCO API(可选)。如果你想在 COCO 数据集上进行训练和评估,可以安装 COCO API。在命令行中运行以下命令: ``` pip install cython; pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI' ``` 9. 最后,测试安装是否成功。可以运行以下命令执行一个简单的示例: ``` python demo/demo.py ``` 按照以上步骤配置 maskrcnn-benchmark 环境后,你将能够在 Windows 下使用该库进行目标检测和实例分割任务。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值