keras Flatten层在深度学习模型中的放置

taor4pac  于 2023-03-02  发布在  其他
关注(0)|答案(1)|浏览(159)

我做了一个用于分类的深度学习模型。该模型由4个Conv2d层、1个池化层、2个密集层和1个扁平层组成。当我这样安排层时:Conv2D,Conv2D,Conv2D,Conv2D,pooling,dense,flattened,dense我的结果是好的。但是当我按照这个排列时:Conv2D,Conv2D,Conv2D,Conv2D,pooling,flatten,dense,dense那么分类效果不好,我的问题是把flatten层放在两个dense层中间是正确的还是不正确的?
我是否可以遵循获得良好分类结果的层模式?

azpvetkf

azpvetkf1#

通常,不建议在密集层之间夹一个扁平层,正如Corralian所建议的,它没有提供任何价值。您的其他架构Conv2D, Conv2D, Conv2D, Conv2D, pooling, flatten, dense, dense更合法。如果您的模型为您提供了良好的结果,您可能希望保留它,但从技术上讲,您不需要在两个密集层之间夹一个扁平层。
你可以考虑使用Conv2D, Conv2D, Conv2D, Conv2D, Pooling, dense, dense,或者更好的选择是尝试使用你的架构,比如在四个Conv2d层之间添加另一个池化层,如下所示:Conv2D, Conv2D, Pooling, Conv2D, Conv2D, Pooling, flatten, dense, dense,然后继续调整超参数。

相关问题