717 B
717 B
该文件夹存放使用pytorch实现的代码版本
model.py： 是模型文件
train.py： 是调用模型训练的文件
predict.py： 是调用模型进行预测的文件
class_indices.json： 是训练数据集对应的标签文件
data 文件夹里分为了train和val两部分，分别用来训练和验证，只要将完成分类的照片分别放在“orange”或“yellow”子文件夹中即可
注意：现在文件夹中只有少量的示例图片，如果直接拿来训练结果不会很理想，毕竟，我也不好把所有工作都做完了，这样怎么体现你们的工作呢？对吧
训练完别忘了把生成的权重文件复制到控制程序中使用