Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
test		test
train		train
README.md		README.md
mobilenet_1_0_224_tf_no_top.h5		mobilenet_1_0_224_tf_no_top.h5
test_model.py		test_model.py
transfer_learning.py		transfer_learning.py

Repository files navigation

Transfer_Learning

本案例是Yanshee机器人迁移学习课程的一个简单例程。

文件功能说明

其中mobilenet_1_0_224_tf_no_top.h5文件是预训练模型。 train文件夹下面放了三种训练集数据。test文件夹下面是测试图片数据。 transfer_learning.py是训练模型源文件。 test_model.py是测试模型源文件。

训练模型

python transfer_learning.py

测试模型

python test_model.py