Commit Graph
66 Commits
Author SHA1 Message Date
zhangpu 0ba55a974f 增加多模态模型识别功能,增加了GUI界面,很强大,识别在一秒左右。 2025-11-03 11:32:00 +08:00
zhangpu 9e2ff890f4 增加多模态模型识别功能,可以进行识别了,程序跑通了 2025-11-03 10:11:14 +08:00
zhangpu 9e12a5fbc3 增加多模态模型识别功能。 2025-11-03 09:29:57 +08:00
zhangpu 9d8eccde84 修改参数。 2025-10-31 14:16:34 +08:00
zhangpu bce02fc418 Merge remote-tracking branch 'gitee/detached'
# Conflicts:
#	classifier/embedding_food_classifier_app.py
#	faiss_vector_db/build_faiss_index.py
#	faiss_vector_db/visualize_embeddings.py
#	train/train_embedding.py
2025-10-29 17:01:29 +08:00
zhangpu 07bae61a10 代码回滚:增加未知类别负样本之后,模型不收敛,训练效果特别差,现在进行回滚。回滚之后效果非常不错。 2025-10-29 16:40:05 +08:00
zhangpu 519bd1c189 增加未知类别负样本 2025-10-23 16:05:17 +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 e74c42b5d4 调整模型路径。 2025-10-16 14:41:59 +08:00
zhangpu 3fd23dcc7d 增加了向量可视化脚本。 2025-10-13 14:08:59 +08:00
zhangpu 1719a0a961 修改相似图片的路径,这样脚本就可以启动了 2025-10-11 18:38:14 +08:00
zhangpu 93e7b9490a 处理程序在徐总电脑上无法预览图片的Bug. 2025-10-11 18:04:52 +08:00
zhangpu dffe328e55 修复新增菜品不更新的问题 2025-10-11 17:53:31 +08:00
zhangpu b62a64033a 修改配置路径 2025-10-11 17:39:57 +08:00
zhangpu 2660d503d6 只是更改了一些配置 2025-10-10 15:59:01 +08:00
zhangpu 9eed8acb71 确保图片压缩都采用F.interpolate(),这样和安卓端得到的结果基本差不多。 2025-10-10 10:58:59 +08:00
zhangpu 86a4f1ff67 增加TODO 2025-10-09 18:05:43 +08:00
zhangpu 7faf3f6480 做了一个向量相乘的APP,避免embedding_mobile递归循环. 2025-10-09 17:58:06 +08:00
zhangpu 065192fff8 增加了导出安卓的一层网络,可以避免安卓使用向量数据库。 2025-10-09 13:29:53 +08:00
zhangpu a14aa7f8e1 修改配置 2025-10-09 09:59:36 +08:00
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