我正在试验一个预先训练过的模型(例如VGG,AlexNet等)。在从另一个模型输出的嵌入的顶部。我认为,对我来说唯一不清楚的部分是,我将如何使输入维度与新增加的预培训模型一起工作?更具体地说:
classification
的嵌入。
是否有任何方法可以让它工作,这样我就可以输入一个已经处理过的图像嵌入到另一个预先训练过的模型中,并成功地执行图像分类?
发布于 2021-09-20 13:32:18
如果你已经从你的图像中提取了一个嵌入,你就不应该使用CNN。一个典型的CNN结构由特征提取器(卷积层)和分类器(完全连接层)组成。卷积部分的目的是从图像中提取相关信息,而后一部分映射这些信息以完成期望的任务(例如,分类任务)。
在您的情况下,使用一个完全连接的层作为model 2
是有意义的。
https://stackoverflow.com/questions/69253628
复制相似问题