解决:修改model.py的
anchors = KL.Lambda(lambda x: tf.Variable(anchors), name="anchors")(input_image)
为:
anchor_layer = AnchorsLayer(name="anchors")
anchors = anchor_layer(anchors)
并在此之前加入函数定义:
class AnchorsLayer(tf.keras.layers.Layer):
def __init__(self, name="anchors", **kwargs):
super(AnchorsLayer, self).__init__(name=name, **kwargs)
def call(self, anchor):
return anchor
def get_config(self) :
config = super(AnchorsLayer, self).get_config()
return config
参考:https://github.com/matterport/Mask_RCNN/issues/1930
原因:mask_rcnn_coco.h5不完整
解决:重新下载mask_rcnn_coco.h5文件,点击下载