-
基于keras的转移学习模块transfer_learning,适用于任意图像分类任务
资源介绍
转移学习
概述
transfer_learning是用于任意端到端图像分类的基于的转移学习模块。 该模块支持基于以下任何预训练模型的转移学习
转移学习通过以下两个步骤进行
通过*模型快速学习*模型是经过预训练的模型,具有完全连接层的最后一块被重新训练以适应特定的分类任务。
微调未冻结层精调首先将其权重初始化为第一步训练的*模型的权重,然后重新训练那些未冻结的层,通常是最终的完全连接的层块及其先前相邻的卷积层块。
有关*模型和微调过程的详细信息,请参阅Francois Chollet的出色。
设置
将模块的所有Python文件下载到项目文件夹中。 为了使模块正确找到数据,需要以下文件夹结构来存储数据
project_dir/
data/
train/
category_1
image_11