多应用+插件架构,代码干净,二开方便,首家独创一键云编译技术,文档视频完善,免费商用码云13.8K 广告
**预训练网络**(pretrained network): * 是一个保存好的网络,之前已在大型数据集(通常是大规模图像分类任务)上训练好。训练了一个网络(其类别主要是**动物和日常用品**),然后将这个训练好的网络应用于某个**不相干**的任务,比如在图像中**识别家具**。 ***** ### **特征提取** 特征提取是使用之前网络学到的表示来从新样本中提取出有趣的特征。然后将这些特征输入一个新的分类器,从头开始训练。 图像分类的**卷积神经网络**包含两部分: * 一系列池化层和卷积层:模型的**卷积基**(convolutional base) * 一个密集连接分类器 对于**卷积神经网络**而言,**特征提取**就是取出之前训练好的网络的卷积基,在上面运行新数据,然后在输出上面训练一个新的分类器。 ![](https://img.kancloud.cn/95/80/95803910c975c2e2efd9853cdb443903_615x418.png) * 某个卷积层提取的表示的**通用性**(以及可复用性)取决于该层在模型中的**深度** * 新数据集与原始模型训练的数据集有**很大差异**,那么最好只使用模型的**前几层**来做特征提取,而不是使用整个卷积基 * 模型中更靠近底部的层提取的是局部的、高度通用的特征图(比如视觉边缘、颜色和纹理),而更靠近顶部的层提取的是更加抽象的概念(比如“猫耳朵”或“狗眼睛”) ### **微调模型** * 与特征提取互为补充。对于用于特征提取的冻结的模型基,微调是指将其顶部的几层“解冻”,并将这解冻的几层和新增加的部分联合训练。 * **微调**,它只是略微调整了所复用模型中更加抽象的表示,以便让这些表示与手头的问题更加相关。 #### **步骤:** 1. 在已经训练好的基网络(base network)上添加自定义网络。 2. 冻结基网络。 3. 训练所添加的部分。 4. 解冻基网络的一些层。 5. 联合训练解冻的这些层和添加的部分。 * [ ] 微调最后三个卷积层`block5_conv1`、`block5_conv2`和`block5_conv3`