Mask-RCNN代码Mask_RCNN/samples路径下有一个demo.ipynb的文件就是用来测试的,所以我们在这个基础上更改一下,其实主要就是我们新建一个test.ipynb,然后把demo.ipynb代码复制过来,根据需要更改。
一、将demo代码更改
将demo中的代码
COCO_MODEL_PATH = os.path.join(ROOT_DIR, "mask_rcnn_coco.h5")
更改为:这里面的h5文件是我们训练的结果,
COCO_MODEL_PATH = os.path.join(ROOT_DIR, "mask_rcnn_shapes_0005.h5")
demo中代码class InferenceConfig(coco.CocoConfig):也需要进行更改,要与自己的config匹配。
class InferenceConfig(shapeconfig.ShapesConfig):
ShapesConfig类中代码:
from mrcnn.config import Config
class ShapesConfig(Config):
"""Configuration for training on the toy shapes dataset.
Derives from the base Config class and overrides values specific
to the toy shapes dataset.
"""
# Give the configuration a recognizable