首页 > 解决方案 > 训练 CNN 模型时如何实现自定义过滤器?

问题描述

我正在制作一个 CNN 模型,并希望通过在图像上应用自定义过滤器来处理图像,然后再将它们输入原始 CNN。知道如何实现吗?我已经尝试过以下链接中提到的方法,但它会引发“断言错误”! 如何在基于 Keras 的 CNN 中包含自定义过滤器?

在此处输入图像描述 在此处输入图像描述

标签: pythonimage-processingkerasconv-neural-network

解决方案


Assertion Error当处理的输出维数与原始 CNN 的输入维数不匹配时,通常会出现此问题。

确保在提到的链接中生成的输出尺寸与原始 CNN 的尺寸相同。

它应该工作!希望能帮助到你 :)


推荐阅读