devbox
代码探险家
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

Mask Rcnn目标分割-项目搭建及跑通测试代码_maskrcnn h5

作者：代码探险家 | 2024-08-07 09:27:29

赞

踩

maskrcnn h5

本文介绍了Mask Rcnn目标分割项目的搭建及运行过程，并对搭建过程中可能出现的问题进行了解答。

环境：Cuda10.2+tensorflow-gpu1.13.2

Mask R-CNN是一个实例分割算法，可以用来做“目标检测”、“目标实例分割”、“目标关键点检测”。

一、下载项目源码

https://github.com/matterport/Mask_RCNN/releases下载最新版本

Source code (zip)：源码压缩包下载。

balloon_dataset.zip：气球数据集，里面的图片并不是很多，几十张。

mask_rcnn_balloon.h5：训练好的气球数据集权重。

mask_rcnn_coco.h5：基于mask rcnn训练好的coco数据集的权重，这个要感谢网友分享的下载链接，这里附上下载地址

mask_rcnn_coco.h5：下载链接如下：

链接：https://pan.baidu.com/s/1Mj9xOj0o8z3Rn-NBo6TRgw 
提取码：l3pb

二、需要安装的环境：

numpy

scipy

Pillow

cython

matplotlib

scikit-image

tensorflow>=1.3.0

keras>=2.0.8

opencv-python

h5py

imgaug

关于CUDA+tensorflow-gpu的安装可以参考我的另一篇博文

Windows10下安装Anaconda、Pycharm、Tensorflow-gpu，并在Pycharm中配置Tensorflow_肯定有人改了我的代码的博客-CSDN博客https://blog.csdn.net/qq_40770527/article/details/108349891?spm=1001.2014.3001.5501

三、测试

1、Windows下安装cocoapi方法

由于源码基于coco数据集进行的测试，检测代码中包含一些coco相关的模块，要正常运行代码，需要先安装cocoapi，具体步骤如下：

（1）在  Github下载源码并进行解压；https://github.com/philferriere/cocoapi

（2）安装 Visual C++ 2015；http://go.microsoft.com/fwlink/?LinkId=691126（因为我的电脑早就安装了VS，所以这一步骤我直接跳过了）

（3）进入coco源码setup.py所在目录 cocoapi-master\PythonAPI；

（4）执行命令：python setup.py build_ext install（注意是在PythonAPI目录）

2、读取摄像头实时检测

Mydemo_came.py

将mask_rcnn_coco.h5权重文件放到项目文件下

  1. import os
  2. import sys
  3. import random
  4. import math
  5. import numpy as np
  6. import skimage.io
  7. import matplotlib
  8. import matplotlib.pyplot as plt
  9. import cv2
  10. import time
  11. # Root directory of the project
  12. ROOT_DIR = os.path.abspath("./")
  13.  
  14. # Import Mask RCNN
  15. sys.path.append(ROOT_DIR)  # To find local version of the library
  16. from mrcnn import utils
  17. import mrcnn.model as modellib
  18. from mrcnn import visualize
  19. # Import COCO config
  20. sys.path.append(os.path.join(ROOT_DIR, "samples/coco/"))  # To find local version
  21. import coco
  22.  
  23.  
  24. # Directory to save logs and trained model
  25. MODEL_DIR = os.path.join(ROOT_DIR, "logs")
  26.  
  27. # Local path to trained weights file
  28. COCO_MODEL_PATH = os.path.join(MODEL_DIR ,"mask_rcnn_coco.h5")
  29. # Download COCO trained weights from Releases if needed
  30. if not os.path.exists(COCO_MODEL_PATH):
  31.     utils.download_trained_weights(COCO_MODEL_PATH)
  32.     print("***********************")
  33.  
  34. # Directory of images to run detection on
  35. IMAGE_DIR = os.path.join(ROOT_DIR, "images")
  36.  
  37. class InferenceConfig(coco.CocoConfig):
  38.     # Set batch size to 1 since we'll be running inference on
  39.     # one image at a time. Batch size = GPU_COUNT * IMAGES_PER_GPU
  40.     GPU_COUNT = 1
  41.     IMAGES_PER_GPU = 1
  42.  
  43.  
  44. config = InferenceConfig()
  45. config.display()
  46.  
  47.  
  48. # Create model object in inference mode.
  49. model = modellib.MaskRCNN(mode="inference", model_dir=MODEL_DIR, config=config)
  50.  
  51. # Load weights trained on MS-COCO
  52. model.load_weights(COCO_MODEL_PATH, by_name=True)
  53.  
  54. # COCO Class names
  55. # Index of the class in the list is its ID. For example, to get ID of
  56. # the teddy bear class, use: class_names.index('teddy bear')
  57. class_names = ['BG', 'person', 'bicycle', 'car', 'motorcycle', 'airplane',
  58.                'bus', 'train', 'truck', 'boat', 'traffic light',
  59.                'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird',
  60.                'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant', 'bear',
  61.                'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie',
  62.                'suitcase', 'frisbee', 'skis', 'snowboard', 'sports ball',
  63.                'kite', 'baseball bat', 'baseball glove', 'skateboard',
  64.                'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup',
  65.                'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple',
  66.                'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza',
  67.                'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed',
  68.                'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote',
  69.                'keyboard', 'cell phone', 'microwave', 'oven', 'toaster',
  70.                'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors',
  71.                'teddy bear', 'hair drier', 'toothbrush']
  72. # Load a random image from the images folder
  73. #file_names = next(os.walk(IMAGE_DIR))[2]
  74. #image = skimage.io.imread(os.path.join(IMAGE_DIR, random.choice(file_names)))
  75. cap = cv2.VideoCapture(0)
  76.  
  77. while(1):
  78.     # get a frame
  79.     ret, frame = cap.read()
  80.     # show a frame
  81.     start =time.clock()
  82.     results = model.detect([frame], verbose=1)
  83.     r = results[0]
  84.  
  85.     visualize.display_instances(frame, r['rois'], r['masks'], r['class_ids'],
  86.                             class_names, r['scores'])
  87.  
  88.     cv2.imshow("capture", frame)
  89.     end = time.clock()
  90.     print(end-start)
  91.     if cv2.waitKey(1) & 0xFF == ord('q'):
  92.         break
  93.  
  94. cap.release()
  95. cv2.destroyAllWindows()
  96.  
  97. #image= cv2.imread("C:\\Users\\18301\\Desktop\\Mask_RCNN-master\\images\\9.jpg")
  98. ## Run detection
  99. #
  100. #results = model.detect([image], verbose=1)
  101. #
  102. #print(end-start)
  103. ## Visualize results
  104. #r = results[0]
  105. #visualize.display_instances(image, r['rois'], r['masks'], r['class_ids'],
  106. #                            class_names, r['scores'])

结果如图所示：

3、检测单张图片

可随机检测文件夹内某个图片或指定图片路径进行检测两种方式

Mydemo_pic.py

  1. import os
  2. import sys
  3. import random
  4. import math
  5. import numpy as np
  6. import skimage.io
  7. import matplotlib
  8. import matplotlib.pyplot as plt
  9. import cv2
  10. import time
  11. # Root directory of the project
  12. ROOT_DIR = os.path.abspath("./")
  13.  
  14. # Import Mask RCNN
  15. sys.path.append(ROOT_DIR)  # To find local version of the library
  16. from mrcnn import utils
  17. import mrcnn.model as modellib
  18. from mrcnn import visualize
  19. # Import COCO config
  20. sys.path.append(os.path.join(ROOT_DIR, "samples/coco/"))  # To find local version
  21. import coco
  22.  
  23.  
  24. # Directory to save logs and trained model
  25. MODEL_DIR = os.path.join(ROOT_DIR, "logs")
  26.  
  27. # Local path to trained weights file
  28. COCO_MODEL_PATH = os.path.join(MODEL_DIR ,"mask_rcnn_coco.h5")
  29. # Download COCO trained weights from Releases if needed
  30. if not os.path.exists(COCO_MODEL_PATH):
  31.     utils.download_trained_weights(COCO_MODEL_PATH)
  32.     print("***********************")
  33.  
  34. # Directory of images to run detection on
  35. IMAGE_DIR = os.path.join(ROOT_DIR, "images")
  36.  
  37. class InferenceConfig(coco.CocoConfig):
  38.     # Set batch size to 1 since we'll be running inference on
  39.     # one image at a time. Batch size = GPU_COUNT * IMAGES_PER_GPU
  40.     GPU_COUNT = 1
  41.     IMAGES_PER_GPU = 1
  42.  
  43.  
  44. config = InferenceConfig()
  45. config.display()
  46.  
  47.  
  48. # Create model object in inference mode.
  49. model = modellib.MaskRCNN(mode="inference", model_dir=MODEL_DIR, config=config)
  50.  
  51. # Load weights trained on MS-COCO
  52. model.load_weights(COCO_MODEL_PATH, by_name=True)
  53.  
  54. # COCO Class names
  55. # Index of the class in the list is its ID. For example, to get ID of
  56. # the teddy bear class, use: class_names.index('teddy bear')
  57. class_names = ['BG', 'person', 'bicycle', 'car', 'motorcycle', 'airplane',
  58.                'bus', 'train', 'truck', 'boat', 'traffic light',
  59.                'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird',
  60.                'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant', 'bear',
  61.                'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie',
  62.                'suitcase', 'frisbee', 'skis', 'snowboard', 'sports ball',
  63.                'kite', 'baseball bat', 'baseball glove', 'skateboard',
  64.                'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup',
  65.                'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple',
  66.                'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza',
  67.                'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed',
  68.                'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote',
  69.                'keyboard', 'cell phone', 'microwave', 'oven', 'toaster',
  70.                'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors',
  71.                'teddy bear', 'hair drier', 'toothbrush']
  72.  
  73. # Load a random image from the images folder
  74. file_names = next(os.walk(IMAGE_DIR))[2]
  75. image = skimage.io.imread(os.path.join(IMAGE_DIR, random.choice(file_names)))
  76.  
  77. # Run detection
  78. results = model.detect([image], verbose=1)
  79.  
  80. # Visualize results
  81. r = results[0]
  82. visualize.display_instances(image, r['rois'], r['masks'], r['class_ids'],
  83.                             class_names, r['scores'])
  84.  
  85.  
  86.  
  87. #image= cv2.imread("C:\\Users\\18301\\Desktop\\Mask_RCNN-master\\images\\9.jpg")
  88. ## Run detection
  89. #
  90. #results = model.detect([image], verbose=1)
  91. #
  92. #print(end-start)
  93. ## Visualize results
  94. #r = results[0]
  95. #visualize.display_instances(image, r['rois'], r['masks'], r['class_ids'],
  96. #                            class_names, r['scores'])

测试结果如图所示：

四、可能出现的报错及解决方法

注意事项：这里首先总结下我在跑代码时遇到的主要问题及解决思路。

（1）由于我之前装了tensorflow2.*最新版本，而原代码是在tensorflow1.3环境下搭建的，所以请中间报的很多错误都是由于版本问题导致的，中间我也尝试过在不降低tensorflow版本的前期下解决问题，结果发现问题一个接着一个，解决了这个马上遇到新的问题，最后没办法只能选择降低tensorflow版本

（2）在安装tensorflow低版本时又遇到了很多问题，主要原因在于我之前安装了很多最新的包，而tensorflow1.3所依赖的一些包可能版本又很低，导致中间出现各种报错，这里建议使用conda install tensorflow***代替pip install tensorflow***（虽然我之前一直用的pip install 也都没有问题，但这次确实是只有conda install 才解决了问题，详细原因也不太清楚，可能是conda安装时会自动添加相关依赖）

（3）如果还是出现一些相关包的功能报错，极大的可能是由于版本问题，最好的方法就是查找该包与tensorflow1.3对应的版本，降低该包当前版本。

具体报错问题：

1、一般报错缺少某个包，直接pip install ***安装就可以

ModuleNotFoundError: No module named 'Cython'

ModuleNotFoundError: No module named 'imgaug'

2、AttributeError: module 'tensorflow' has no attribute 'log'

因为我一开始用的tensorflow2.0以上版本，而原代码使用的tensorflow1.3，导致出现了各种问题，

如果是2.0的版本请修改为更低的版本试试，比如1.4或者以下版本

3、from tensorflow.python.eager.context import get_config报错

原因是，keras和tensorflow版本不匹配的问题，我的就是keras直接安装，结果安装的最新版本，和tensorflow不匹配

最后，无赖降低了版本，

4.ImportError: DLL load failed: 找不到指定的模块。

Failed to load the native TensorFlow runtime.

网上有说是protobuf问题。 在自动安装了tensorflow时， protobuf安装的是最新版本3.20.0， 出现了不兼容的问题。更换为 protobuf 3.6.0即可

pip install protobuf==3.6.0

弄了半天还是报错，后来看到说大概应该是因为缺少依赖，还有的相似的问题，可能是因为tensorflow和你的cuda版本不匹配。比如我的电脑上cuda就是10.2版本的，但是需要的tensorflow版本只支持到cuda10.0。如果你用pip安装，这种情况下你就要手动重新配cuda，但是用conda安装，他会自动帮你把需要的cudatoolkit一起安装上。这是用conda安装时，自动装下来的包有cudatoolkit。

所以装这些包的时候能用conda装尽量用conda装。conda install会帮你把需要的依赖一起安装好，而pip install只会帮你装一个你指定的包。

解决方法:

Pip uninstall tensorflow

Conda install tensorflow==1.13.2

5、File "h5py\h5.pyx", line 1, in init h5py.h5

ImportError: DLL load failed: 找不到指定的程序。

AttributeError: module 'h5py' has no attribute 'File'

由于h5py库的更新，安装过程中会自动安装h5py=3.0.0以上的版本，会导致decode(“utf-8”)的错误！

各位一定要在安装完tensorflow后利用命令装h5py=2.10.0！

6、AttributeError: type object 'h5py.h5.H5PYConfig' has no attribute '__reduce_cython__'

查看原因说可能存在多个h5py版本，

于是我分别使用conda uninstall

和pip uninstall 卸载了h5py，并在安装目录中删掉了所有h5py包

最后再装了一次，解决了问题

7、报错信息：UserWarning: Matplotlib is currently using agg, which is a non-GUI backend, so cannot show the figure.

网上试了很多方法，有说要加use的，但是我试了都没用

import matplotlib

matplotlib.use('Qt5Agg')#必须显式指明matplotlib的后端

import matplotlib.pyplot as plt

后来我想到可能还是版本过高的问题，我用的是3.版本，于是我卸载了重装了2.版本，解决了问题

8、报错信息：Input image dtype is bool. Interpolation is not defined with bool data type

原因：可能是scikit-image的版本太高，与其他环境不匹配，我卸载当前版本后安装了一个低版本的，解决了问题

解决方案：

pip uninstall scikit-image

pip install scikit-image==0.16.2

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/代码探险家/article/detail/941921
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号