Windows搭建MaskRCNN环境

最新推荐文章于 2025-06-01 20:25:54 发布

南方姑娘-qing

最新推荐文章于 2025-06-01 20:25:54 发布

阅读量646

点赞数 5

CC 4.0 BY-SA版权

分类专栏： MaskRCNN 文章标签： MaskRCNN

本文链接：https://blog.youkuaiyun.com/csdn_cai_csdn/article/details/144031767

环境：python3.6

1. 在miniconda上创建虚拟环境

miniconda下载地址：https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda

# 创建环境
conda create -n maskrcnn python=3.6
# 激活 maskrcnn 环境，后续操作都在这个环境下进行
conda activate maskrcnn

2. 升级pip

python -m ensurepip
python -m pip install --upgrade pip

3. 下载Mask_RCNN代码

对于MaskRCNN2.1版本直接下载并将权重文件mask_rcnn_coco.h5放在代码的根目录下即可，对于2.0版本需要在powershell中找到文件根目录输入下面命令进行安装，区别在于有没有setup.py文件

源码地址：https://github.com/matterport/Mask_RCNN/releases
进入项目根目录：

python setup.py install

在这里插入图片描述
mask_rcnn_coco.h5模型放在项目根目录下

4. 安装插件库

在maskRCNN2.0里面有一个requirements.txt文件，可以直接按照这个安装

pip install -r requirements.txt # 大概率会失败

若安装报错，按照如下版本手动安装

pip install numpy==1.17.0
pip install scipy==1.2.1
pip install Pillow==8.4.0
pip install cython==0.29.28
pip install matplotlib==3.3.4
pip install scikit-image==0.17.2
pip install keras==2.1.6
pip install opencv-python==4.3.0.38
pip install h5py==2.10.0
pip install imgaug==0.4.0
pip install ipython==7.16.3

5. 安装pycocotools

# windows环境
pip install pycocotools-windows

6. 测试COCO数据集

在根目录下创建mode_coco.py

'''
Author: qingqingdan 1306047688@qq.com
Date: 2024-11-22 17:48:33
LastEditTime: 2024-11-29 15:32:37
Description: 用coco模型去预测图片
'''
import os
import sys
import cv2
import numpy as np
import matplotlib.pyplot as plt
import colorsys
from skimage.measure import find_contours

from PIL import Image

# 将 Mask R-CNN 模块路径添加到系统路径
ROOT_DIR = os.path.abspath("./")  # 替换为 Mask R-CNN 文件夹路径
sys.path.append(ROOT_DIR)

from mrcnn.config import Config
from mrcnn import model as modellib
from mrcnn import visualize
from mrcnn.model import log

# 加载 COCO 类别信息
from mrcnn.config import Config
from mrcnn.utils import Dataset

# COCO 类别
COCO_CLASS_NAMES = [<