Commit graph

57 commits

Author SHA1 Message Date
weikaiwen348-code
37e24a0b09 Merge branch 'data_cleaning_test' into main 2026-06-03 20:53:07 +08:00
weikaiwen348-code
be45ad32eb 新增web推理前端:Gradio网页界面,支持上传图片实时分类 2026-06-03 20:52:28 +08:00
weikaiwen348-code
0e9bd58dce 重置所有文件与main一致,新增app.py(Gradio推理前端) 2026-06-03 20:33:22 +08:00
7ebe975186
Update README.md 2026-05-18 22:41:21 +08:00
1557a2c6a0 Update training_log.csv
不小心上传空白csv,现已恢复
2026-05-17 20:22:53 +08:00
8008f2da8f Update README.md
修改Readme.md
2026-05-17 19:57:19 +08:00
52b03e981c README: 补充 PyTorch CUDA/XPU 版本安装提示 2026-05-17 19:52:38 +08:00
ed2778972b 添加精简版 requirements.txt,更新 README 环境要求 2026-05-17 19:51:50 +08:00
65c9742a1c 修改accuracy_bar.png 2026-05-17 19:46:31 +08:00
25e11c1914 完善 README:新增基线对比章节,嵌入 baseline 对比图,更新文件说明和目录结构 2026-05-17 19:41:30 +08:00
b7341c746f 调整准确率柱状图 y 轴范围,放大模型间差异 2026-05-17 19:34:46 +08:00
307b66e9db fix: 移除旧版 sklearn 不兼容的 multi_class 参数 2026-05-17 18:47:14 +08:00
562fc4142a 新增 HOG + LogisticRegression 纯传统基线模型(零神经网络依赖) 2026-05-17 18:43:28 +08:00
3fee1c82ab compare_models.py: 新增 PR 曲线对比图 2026-05-17 18:31:32 +08:00
010dacb533 新增 ResNet34 ImageNet 预训练 + 10% 数据微调模型 (baseline) 2026-05-17 17:20:09 +08:00
547d96cfa9 compare_models.py: 添加 CSV 缓存机制,已有预测数据时跳过重复计算 2026-05-17 16:55:27 +08:00
818d98d06c 重构:将 Baseline.py 迁移至 baseline/ 目录,新增多模型对比脚本 2026-05-17 16:14:48 +08:00
3624f058c2 添加 VGG16 + KNN 基线模型 Baseline.py 2026-05-14 19:50:05 +08:00
76b56dd64b 在 README 中直接展示评估与训练曲线图片 2026-05-14 16:04:31 +08:00
44c70250e3 更新 AGENTS.md、README.md,添加可视化脚本与输出文件到版本控制 2026-05-14 16:02:47 +08:00
2f4e9df26e Merge branch 'Resnet34-test' 2026-05-14 15:51:52 +08:00
b4c99489b3 Update Evaluate.py
修改曲线绘画函数
2026-05-14 15:49:52 +08:00
ywd09
ce0c6da36a 模型评估程序 2026-05-14 00:37:04 +08:00
4575f3390f refactor: replace custom Bottleneck model with standard ResNet-34 + Dropout 2026-05-12 15:56:28 +08:00
cb17be247e 添加日志输出功能 2026-05-03 16:58:13 +08:00
c534eef76d Delete finetune.py 2026-04-30 15:28:15 +08:00
c2a82ad3ba 规范化重命名 2026-04-30 15:27:06 +08:00
971287878-art
d11f6266a7 提交模型微调代码 2026-04-25 22:47:16 +08:00
971287878-art
45747e40e7 Create finetune.py
添加模型微调代码
2026-04-25 21:22:51 +08:00
808191b843 完善项目README文档,补充架构说明与使用指南 2026-04-25 12:45:00 +08:00
6f54c8b13e 改用宏F1评估与类别加权损失 2026-04-25 12:17:26 +08:00
f8bb340a70 添加模型断点继续加载训练功能 2026-04-21 15:57:13 +08:00
968e108857 提高数据加载鲁棒性 2026-04-16 20:57:48 +08:00
cff98f70cc Update Train.py
添加模型中断后继续训练的功能
2026-04-16 20:43:08 +08:00
cb6bdc7eb8 模型提升到50层 2026-04-16 14:12:25 +08:00
b2f7a9c172 模型规范设计 2026-04-16 13:55:02 +08:00
793852eedd Merge branch 'data_cleaning_test' 2026-04-13 22:20:54 +08:00
1350cdd319 数据清理程序改成相对路径 完成dataloader 2026-04-13 22:20:28 +08:00
weikaiwen348-code
543e833fd0 代码路径调整 2026-04-13 16:44:23 +08:00
weikaiwen348-code
934b1d2a7d Update merge_classes.py
完成:编写 265 类至 4 大类的数据集自动化合并脚本。
2026-04-13 16:26:26 +08:00
weikaiwen348-code
3809899218 Merge branch 'main' into data_cleaning_test 2026-04-13 15:48:15 +08:00
e380094fa1 修改dataloader 2026-04-12 20:10:53 +08:00
af1bed54ea 模型规范重命名 2026-04-12 14:39:58 +08:00
39e8781bdd 一份数据加载用的代码,没有调整,因为现在还没有配置好数据集 2026-04-12 14:39:44 +08:00
weikaiwen348-code
4a00bd43b4 Create merge_classes.py
自动生成类别映射并合并四大类数据集的脚本(ai生成还未校验)
2026-04-12 13:55:41 +08:00
2909902595 修改模型 2026-04-10 21:04:09 +08:00
6d4e86714e 添加训练文件(目前仍然不可用⚠️ 2026-04-10 17:16:49 +08:00
0dc8aba668 Merge branch 'dev' 2026-04-10 17:04:53 +08:00
a16b691244 Update model.py
初步完成模型构建
2026-04-10 17:04:26 +08:00
5d51c20a1d Merge branch 'dev' 2026-04-10 16:12:16 +08:00