Commit Graph
44 Commits
Author SHA1 Message Date
zhangpu f3e3bd048a 增加了相似菜品预览功能,更完整了,交互更方便了。 2025-09-29 18:03:28 +08:00
zhangpu 1f04bb5c01 增加了新增菜品功能。 2025-09-29 14:16:50 +08:00
zhangpu 1308c42cb5 增加向量模型转化成安卓可运行模型的程序 2025-09-29 09:41:15 +08:00
zhangpu 1bb60e9f90 增加注释,去掉dropout 2025-09-26 11:32:38 +08:00
zhangpu 97e78e65f0 删除文件 classifier/embedding_foood_classifier_app.py 2025-09-23 07:36:08 +00:00
zhanghuan d51853997f 建立FAISS索引。 2025-09-23 15:34:23 +08:00
zhanghuan 2d5096a0b2 修改可视化客户端ImageTk为ctk,消除警告。 2025-09-22 14:09:19 +08:00
zhanghuan 8791b4f2c0 pretrained=pretrained,解决预训练模型API变化导致的警告。 2025-09-22 13:59:05 +08:00
zhanghuan 13b4e83019 增加embedding方式,可视化的推理界面。 2025-09-22 13:40:49 +08:00
zhanghuan 8d6555747b 增加FAISS向量数据库对所有训练图片进行向量化的过程。 2025-09-22 11:29:22 +08:00
zhanghuan be284442e4 增加FAISS向量数据库检索存储等功能。 2025-09-22 10:42:57 +08:00
zhanghuan e34ecbd94d 增加相应注释! 2025-09-19 09:18:35 +08:00
zhanghuan 7edbf62e26 增加了embedding模型训练的代码 2025-09-17 14:57:33 +08:00
zhanghuan 8818643025 增加resnet_embedding的代码 2025-09-17 14:44:34 +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 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 06e440a6d1 修复toAndroid.py,避免递归调用。 2025-09-11 14:41:30 +08:00
zhanghuan 2f6ce5ffc3 在模型定义中,增加了移动端的前向传播过程。修改了toAndroid的代码。 2025-09-11 14:35:46 +08:00
zhanghuan 4cfbfecf9d 推理阶段不需要做图片预处理,直接扔到模型里面就好了。 2025-09-11 14:23:45 +08:00
zhanghuan d9280fd0a0 训练的时候缩放,推理的时候,在模型中缩放! 2025-09-11 14:13:44 +08:00
zhanghuan ee829dc8c9 在模型中增加预处理步骤! 2025-09-11 13:52:36 +08:00
zhanghuan adefc3ecb2 菜品识别程序,增加向量输出到控制台的功能! 2025-09-11 13:39:49 +08:00
zhanghuan d9057edbd9 提交__init__文件,和网络文件 2025-09-10 15:35:41 +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 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 068c6977c9 修复了预览窗口显示层级的问题,标题增加“数字味道”! 2025-09-03 12:04:26 +08:00
zhanghuan 0467351fde 增加了训练时间! 2025-09-03 11:53:14 +08:00
zhanghuan 6295ec5c1c 根据识别结果来决定识别结果的显示颜色!红色,错误,绿色,正确! 2025-09-03 11:45:48 +08:00
zhanghuan 3aabf88b1f 根据路径自动判断识别结果是否正确! 2025-09-03 11:39:39 +08:00
zhanghuan c2186b8381 将识别结果的字号放大,并且加粗! 2025-09-03 11:29:26 +08:00
zhanghuan 10dc7ec5bd 更新依赖库! 2025-09-03 11:15:12 +08:00
zhanghuan 0496ada9f0 目前可以正常识别了,支持多张图片同时识别! 2025-09-03 11:11:44 +08:00
zhanghuan a60cd58a0b 优化曲线的绘制,避免中文不显示。 2025-09-03 09:57:11 +08:00
zhanghuan 1869fdd3ef 可以实现二分类任务了(西红柿鸡蛋和回锅肉),并且可以打印训练日志图,在验证集上可以达到95%的精度! 2025-09-02 17:11:30 +08:00
zhanghuan b47208c6c0 创建项目,第一次提交 2025-09-02 16:53:04 +08:00