文件名称:transfer_learning:基于keras的转移学习模块,用于任意图像分类
文件大小:12KB
文件格式:ZIP
更新时间:2024-06-03 02:06:01
Python
转移学习 概述 transfer_learning是用于任意端到端图像分类的基于的转移学习模块。 该模块支持基于以下任何预训练模型的转移学习 转移学习通过以下两个步骤进行 通过*模型快速学习*模型是经过预训练的模型,具有完全连接层的最后一块被重新训练以适应特定的分类任务。 微调未冻结层精调首先将其权重初始化为第一步训练的*模型的权重,然后重新训练那些未冻结的层,通常是最终的完全连接的层块及其先前相邻的卷积层块。 有关*模型和微调过程的详细信息,请参阅Francois Chollet的出色。 设置 将模块的所有Python文件下载到项目文件夹中。 为了使模块正确找到数据,需要以下文件夹结构来存储数据 project_dir/ data/ train/ category_1 image_11
【文件预览】:
transfer_learning-master
----config.py(4KB)
----transfer_learning.py(5KB)
----data.py(15KB)
----utils.py(8KB)
----__init__.py(1B)
----models.py(14KB)
----LICENSE(1KB)
----README.md(2KB)