zhangpu
|
25f8e0a267
|
修改训练代码,兼容三个模型的训练。
|
2025-10-16 15:26:58 +08:00 |
|
zhanghuan
|
264d01e9b9
|
增加了数据增强离线的程序(可扩充数据集),将图片压缩从32*32,调整为224*224.
|
2025-09-15 17:26:14 +08:00 |
|
zhanghuan
|
d73331010c
|
将分类的类别也应用在训练程序中,更好的解耦。
|
2025-09-12 11:20:46 +08:00 |
|
zhanghuan
|
da67cb9f4c
|
现在统一使用Torch中的interpolate这个缩放方法,不用torchvision中的transforms的Resize了,应该就可以了。
|
2025-09-11 17:56:25 +08:00 |
|
zhanghuan
|
e1cb936e32
|
屏蔽/255,避免重复除以。
|
2025-09-11 17:49:37 +08:00 |
|
zhanghuan
|
ee829dc8c9
|
在模型中增加预处理步骤!
|
2025-09-11 13:52:36 +08:00 |
|
zhanghuan
|
adefc3ecb2
|
菜品识别程序,增加向量输出到控制台的功能!
|
2025-09-11 13:39:49 +08:00 |
|
zhanghuan
|
97401e908a
|
每次训练完成后,保存训练结果。
|
2025-09-10 15:32:21 +08:00 |
|
zhanghuan
|
5a3fc59573
|
训练指标曲线图配置成setting文件中的变量。
|
2025-09-10 15:19:29 +08:00 |
|
zhanghuan
|
2412c26342
|
settings.py中增加推理模型路径。
|
2025-09-10 15:12:45 +08:00 |
|
zhanghuan
|
06fead1ab3
|
已经把重要的配置全部拎出来了,每次训练只需要修改配置文件就可以了。
|
2025-09-10 15:02:23 +08:00 |
|