flt/ai-titration
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

src

Browse Source 
Former-commit-id: 43bc977a970a5bba09d0afa6f2a85169fe1ed253
This commit is contained in:
flt6
2025-05-15 23:49:13 +08:00
commit 2c2822ff11
720 changed files with 2735 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
Picture_Train/README.md Normal file
View File

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