ASD 项目使用教程

ASD 项目使用教程

1. 项目目录结构及介绍

ASD 项目的目录结构如下：

ASD/
├── configs/
│   ├── asd_drydown_config.yaml
│   └── ...
├── criterions/
│   └── ...
├── datasets/
│   └── ...
├── figs/
│   └── ...
├── metrics/
│   └── ...
├── models/
│   └── ...
├── runners/
│   └── ...
├── utils/
│   └── ...
├── LICENSE
├── README.md
├── requirements.txt
└── run.py

目录介绍

• configs/: 存放项目的配置文件，如 asd_drydown_config.yaml。
• criterions/: 存放模型训练中的损失函数定义。
• datasets/: 存放数据集处理相关的代码。
• figs/: 存放项目中使用的图片文件。
• metrics/: 存放模型评估指标的定义。
• models/: 存放模型的定义和实现。
• runners/: 存放训练和测试的运行脚本。
• utils/: 存放项目中使用的工具函数。
• LICENSE: 项目的开源许可证文件。
• README.md: 项目的介绍文档。
• requirements.txt: 项目依赖的 Python 包列表。
• run.py: 项目的启动文件。

2. 项目启动文件介绍

项目的启动文件是 run.py。该文件用于启动模型的训练和测试过程。

使用方法

python run.py 'config_file_path'

例如，要训练 ASD 模型并将 Agriculture-vison 数据集中的 drydown 类作为正常类，可以使用以下命令：

python run.py /configs/asd_drydown_config.yaml

3. 项目配置文件介绍

项目的配置文件存放在 configs/ 目录下，通常以 .yaml 格式存储。配置文件中包含了模型训练和测试所需的各种参数。

配置文件示例

# asd_drydown_config.yaml
model:
  name: ASD
  parameters:
    learning_rate: 0.001
    batch_size: 32

dataset:
  name: Agriculture-vison
  normal_class: drydown

training:
  epochs: 50
  save_interval: 10

testing:
  visualization: True

配置文件参数说明

• model: 定义模型的名称和参数，如学习率和批量大小。
• dataset: 定义使用的数据集和正常类。
• training: 定义训练的轮数和保存模型的间隔。
• testing: 定义测试时的可视化选项。

通过修改配置文件中的参数，可以调整模型的训练和测试行为。