我做了一个用于分类的深度学习模型。该模型由4个Conv2d层、1个池化层、2个密集层和1个扁平层组成。当我这样安排层时:Conv2D,Conv2D,Conv2D,Conv2D,pooling,dense,flattened,dense我的结果是好的。但是当我按照这个排列时:Conv2D,Conv2D,Conv2D,Conv2D,pooling,flatten,dense,dense那么分类效果不好,我的问题是把flatten层放在两个dense层中间是正确的还是不正确的?
我是否可以遵循获得良好分类结果的层模式?
1条答案
按热度按时间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
,然后继续调整超参数。