Skip to content

Latest commit

 

History

History

tensorflow_classification

该文件夹存放使用tensorflow实现的代码版本

model.py: 是模型文件
train.py: 是调用模型训练的文件
predict.py: 是调用模型进行预测的文件
class_indices.json: 是训练数据集对应的标签文件


若要使用该训练脚本需要下载对应的花分类数据集并将其划分为训练集和验证集。
点击这里会告诉你如何去下载数据集,以及提供了现成的划分数据集脚本