From bf7256032621eac38bd7b242a0d0034e080609fc Mon Sep 17 00:00:00 2001 From: Shaun Zhou Date: Sun, 11 Oct 2026 16:42:13 +0800 Subject: [PATCH] docs: add AGENTS.md for vision model training and evaluation workflow --- AGENTS.md | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 AGENTS.md diff --git a/AGENTS.md b/AGENTS.md new file mode 100644 index 0000000..cecb136 --- /dev/null +++ b/AGENTS.md @@ -0,0 +1,29 @@ + +# Vision Model Training and Evalutation + +本项目目标是构建并训练出在薄膜数据上的缺陷分类定位模型。在聚类和生成 yolo 数据集阶段，我们只考虑基于 convolution 的模型。所有较重的任务均放在远程服务器上运行。 + +## 各目录的作用 + +- configs: 存放相关目标的 yaml 配置文件 +- models: 存放模型的基础骨架 +- datasets: 存放抛出数据样本的 class +- data: 存放数据集或数据集的 link +- scripts: 存放文件类型转换以及批处理脚本 + +## 远程训练机 + +```bash +ssh deep@192.168.31.153 +``` + +## 工作流 + +1. 将无标签数据存放在 `./data/raw/` 下。 +2. 使用 deep model 对无标签数据使用 simclr 进行无监督学习。模型放在 `./runs/` 下。 +3. 用户指定簇数 `k` 进行聚类，结果图片通过硬链接放在另一个 `./runs/` 下。 +4. 用户手动将聚类结果归类到定义的 label 下。可以通过对某一类继续聚类来细化分类。 +5. 用户手动将细化后的分类数据集放到 `./data/` 下。e.g. `./data/orgnized_from_r101_k40/` +6. 将整理好的数据集划分训练集和验证集。使用一个合适的模型（或聚类时的模型）开启训练。 +7. 由用户自行确认模型是否训练完成。 +8. 在训练的模型上，使用 GradCAM 生成 top1-predicted 的热力图放在新的 `./runs/` 下，同时通过框选头部热力区域生成 box。在该目录下简历原始数据集的硬链接，同时每个图片都配上一个 YOLO 格式的数据标签。 \ No newline at end of file