Classification:分类模型在Keras当中的实现
目录
所需环境
tensorflow-gpu==1.13.1
keras==2.1.5
文件下载
训练所需的预训练权重都可以在百度云下载。
链接:
提取码: 36sn
训练所用的示例猫狗数据集也可以在百度云下载。
链接:
提取码: hmmx
预测步骤
a、使用预训练权重
下载完库后解压,model_data已经存在一个训练好的猫狗模型mobilenet025_catvsdog.h5,运行predict.py,输入
img/cat.jpg
b、使用自己训练的权重
按照训练步骤训练。
在classification.py文件里面,在如下部分修改model_path、classes_path、backbone和alpha使其对应训练好的文件;model_path对应logs文件夹下面的权值文件,classes_path是model
1