Commit Graph
36 Commits
Author SHA1 Message Date
zhangpu 40e90ac7c2 参数修改。 2025-12-03 13:43:03 +08:00
zhangpu c74ac621a6 Merge remote-tracking branch 'gitee/master'
# Conflicts:
#	faiss_vector_db/build_faiss_index.py
#	train/grid_search_cosface.py
2025-11-26 14:50:56 +08:00
zhangpu 0e8a1df590 不要类别加权 2025-11-26 14:50:00 +08:00
zhangpu 95d3099cee 修改参数 2025-11-21 10:19:04 +08:00
zhangpu 3ae8c677ff 处理embedding的请求时长。 2025-11-13 16:23:02 +08:00
zhangpu ced0e4f7ef 增加多模态模型开放集识别。 2025-11-13 11:56:51 +08:00
zhangpu b4b2b18ccd 增加网格搜索超参数功能。 2025-11-11 18:03:50 +08:00
zhangpu 78cb1d72e1 修改了验证集正确率问题,之前打印验证集正确率100%,实际测试并不是100% 2025-11-11 14:03:31 +08:00
zhangpu fc594181ed 修改margin参数 2025-11-11 10:42:00 +08:00
zhangpu e267f383d5 给训练脚本,增加早停法。 2025-11-11 09:35:04 +08:00
zhangpu be04d6affb 将margin和scale改为可配置选项(针对不同训练任务) 2025-11-11 09:23:06 +08:00
zhangpu 12e231dc6c 更改配置文件 2025-11-11 09:02:45 +08:00
zhangpu 45ed6b905a 修改配置。 2025-11-07 09:40:54 +08:00
zhangpu a81de1f4ce 增加cosFace训练脚本。 2025-11-05 20:03:55 +08:00
zhangpu f6180ed228 增加配置文件 2025-11-04 14:53:31 +08:00
zhangpu 9d8eccde84 修改参数。 2025-10-31 14:16:34 +08:00
zhangpu 07bae61a10 代码回滚:增加未知类别负样本之后,模型不收敛,训练效果特别差,现在进行回滚。回滚之后效果非常不错。 2025-10-29 16:40:05 +08:00
zhangpu 1c8ceef5b0 修改配置和路径。 2025-10-23 11:08:46 +08:00
zhangpu 5db96600fe 增加一些设置。 2025-10-17 18:24:04 +08:00
zhangpu 25f8e0a267 修改训练代码,兼容三个模型的训练。 2025-10-16 15:26:58 +08:00
zhangpu f3e3bd048a 增加了相似菜品预览功能,更完整了,交互更方便了。 2025-09-29 18:03:28 +08:00
zhanghuan e34ecbd94d 增加相应注释! 2025-09-19 09:18:35 +08:00
zhanghuan 7edbf62e26 增加了embedding模型训练的代码 2025-09-17 14:57:33 +08:00
zhanghuan 44031b92cd 制作了度量学习的开发计划,创建了几个空python文件 2025-09-17 11:41:24 +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 d9280fd0a0 训练的时候缩放,推理的时候,在模型中缩放! 2025-09-11 14:13:44 +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 5f62c6e2e0 在保存模型文件之前,先检查路径是否存在。 2025-09-10 15:05:08 +08:00
zhanghuan 06fead1ab3 已经把重要的配置全部拎出来了,每次训练只需要修改配置文件就可以了。 2025-09-10 15:02:23 +08:00
zhanghuan 4c2fa0e533 把网络模型单独拎出来了,方便解耦。下一步准备把重要的配置全部拎出来。 2025-09-10 14:48:43 +08:00
zhanghuan 393d4c2a60 增加训练时长打印。 2025-09-10 14:21:56 +08:00
zhanghuan 2a3431b286 增加成三分类。 2025-09-10 14:11:11 +08:00
zhanghuan 0496ada9f0 目前可以正常识别了,支持多张图片同时识别! 2025-09-03 11:11:44 +08:00