docs: add AGENTS.md for vision model training and evaluation workflow
This commit is contained in:
@@ -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/<user_defined_name>` 下。
|
||||
3. 用户指定簇数 `k` 进行聚类,结果图片通过硬链接放在另一个 `./runs/<user_defined_name>` 下。
|
||||
4. 用户手动将聚类结果归类到定义的 label 下。可以通过对某一类继续聚类来细化分类。
|
||||
5. 用户手动将细化后的分类数据集放到 `./data/` 下。e.g. `./data/orgnized_from_r101_k40/`
|
||||
6. 将整理好的数据集划分训练集和验证集。使用一个合适的模型(或聚类时的模型)开启训练。
|
||||
7. 由用户自行确认模型是否训练完成。
|
||||
8. 在训练的模型上,使用 GradCAM 生成 top1-predicted 的热力图放在新的 `./runs/<user_defined_name>` 下,同时通过框选头部热力区域生成 box。在该目录下简历原始数据集的硬链接,同时每个图片都配上一个 YOLO 格式的数据标签。
|
||||
Reference in New Issue
Block a user