首页 > 解决方案 > 有谁看到图形断开连接来自哪里?

问题描述

尽管我没有看到断开连接的位置,但我抛出了以下错误。

标签: pythontensorflowkerasneural-network

解决方案


问题在于从基本模型获取层的输出,而在您的模型中没有任何地方使用生成这些输出的基本模型的输入。

这里的问题:base_model.get_layer(concat_with).output

对于您的情况,最简单的解决方案是直接在数据或生成器中使归一化超出模型,并使用:

img_input = base_model.input
encoded = base_model.output

这样你就不需要改变了UpProject。(这可能仍然存在问题,具体取决于所选层是否在基础模型中多次使用)

请注意,如果您希望预训练的基础模型正常工作,您应该对此模型使用正确的预处理:

keras.applications.densenet import preprocess_input
processed_images = preprocess_input(batch_of_images)

如果您使用 PIL 或 Keras 标准生成器加载图像,这将有效。如果您打开cv2,那么您需要先翻转通道。


推荐阅读