文件名称:transfer_learning_tensorflow:使用tensorflow和tensornets的轻松转移学习程序
文件大小:6KB
文件格式:ZIP
更新时间:2024-06-04 09:22:00
Python
transfer_learning_tensorflow A easy transfer learning program using tensorflow and tensornets 用tensorflow和开源库实现迁移学习。 程序主要针对百度点石竞赛商家招牌分类,也可迁移到其他任务。 功能简介: 支持35个模型 可设置随机图像增强 可选择finetune的变量数 可restore保存的ckpt接着训练 说明 更换模型只需替换model.DenseNet201为你想要transfer learn的model,具体支持的模型请移步 请将数据文件放置于datasets文件夹下运行model/train.py即可开始训练 运行model/test.py进行预测 重要参数说明: status: 有两个选项train和restore,分别代表训练和恢复存储的模型继续训练 num_train_v
【文件预览】:
transfer_learning_tensorflow-master
----predict()
--------readme.md(41B)
----model()
--------train.py(10KB)
--------test.py(3KB)
----README.md(2KB)