You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

深度学习:利用预训练网络早期激活层提取特征

领口图像分类任务的特征提取与问题分析

任务概况

  • 数据集:共约25000张图片,划分为14个领口类别(如V领、圆领等)
  • 图像内容:以服装上半部分为核心,部分包含模特面部区域

特征提取方案点评

我认为你选择提取预训练在ImageNet上的VGG16第一个模块后的特征,这个思路非常贴合领口分类的需求——VGG16的早期模块专门负责捕捉线条、边缘、基础形状这类低层次视觉特征,而领口分类恰恰最依赖这些轮廓信息,完全避开了高层特征可能带来的冗余干扰(比如模特面部细节、服装面料纹理等无关信息)。

从常规的VGG16结构来看,第一个模块的组成一般对应model.summary()里的前几层:

卷积层(64个3×3卷积核,ReLU激活)→ 卷积层(64个3×3卷积核,ReLU激活)→ 2×2最大池化层

提取这一模块的输出作为特征,既能精准保留领口的关键轮廓特征,又能控制特征维度,避免后续分类器的训练负担过重。

关于你提到的“总数量”相关问题

你提到“目前存在的问题是总数量……”,内容似乎没补充完整。如果是指数据集类别样本分布不均衡(比如部分领口类别的样本数量远多于其他类别),可以参考这些解决方向:

  • 针对样本量少的类别做针对性数据增强:比如水平翻转、随机裁剪、亮度/对比度微调等,扩充有效样本
  • 使用加权交叉熵损失函数:给样本量少的类别分配更高的损失权重,让模型更关注这类样本的学习
  • 微调VGG16的早期模块:在提取特征的基础上,用你的领口数据集对VGG16的前几个模块做少量微调,让特征更适配你的特定任务

如果是其他和“总数量”相关的问题(比如特征总维度太高、训练集总数据量不足等),可以补充具体细节后再进一步探讨~

内容的提问来源于stack exchange,提问作者user1274878

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.05.19 03:38:44