Mask_RCNN报错合集

最新推荐文章于 2024-11-02 22:51:23 发布

原创最新推荐文章于 2024-11-02 22:51:23 发布 · 1k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #深度学习 #自动驾驶 #神经网络

自动驾驶同时被 2 个专栏收录

3 篇文章

订阅专栏

2 篇文章

订阅专栏

本文解决了两个关于Mask_RCNN模型的问题：一是ValueError错误，通过修改Lambda层为自定义层解决；二是OSError错误，原因是预训练权重文件mask_rcnn_coco.h5不完整，解决方案是重新下载完整的权重文件。

1. ValueError: The following Variables were created within a Lambda layer (anchors)

解决：修改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