代码之家  ›  专栏  ›  技术社区  ›  Prof.Plague

我怎样才能将辍学技术添加到角膜视网膜网?

  •  -2
  • Prof.Plague  · 技术社区  · 7 年前

    我正在使用retinanet-nn模型进行对象检测,我面临过拟合问题。 其中一个解决方案是添加“辍学”。 我用的是喀拉斯密码 Here

    我想将辍学添加到最后一层,但不知道如何添加。

    有人能帮我修改哪个文件吗?如何?

    2 回复  |  直到 7 年前
        1
  •  0
  •   Fabian    7 年前

    您可以尝试将完全连接的层存储到如下变量中:

    fc1 = model.layers[-3]
    fc2 = model.layers[-2]
    predictions = model.layers[-1]
    

    然后创建您的退出层,并重新连接所有层以构建一个新模型,如本文所示: Add dropout layers between pretrained dense layers in keras

    希望这有帮助。

        2
  •  0
  •   Prof.Plague    7 年前

    过了一会儿,我尝试了很多解决方案,但没有一个说怎么添加,所以我尝试了,然后发现如何添加,所以决定自己回答!

    只需要添加这样一行:

    outputs = keras.layers.SpatialDropout1D (rate=dropout_rate) (outputs)

    您可以使用另一种层退出类型,如: SpatialDropout2D 还有更多。