CWT-for-FSS 项目使用教程

1. 项目的目录结构及介绍

CWT-for-FSS/
├── config_files/
│   ├── config1.yaml
│   ├── config2.yaml
│   └── ...
├── doc/
│   ├── README.md
│   └── ...
├── lists/
│   ├── train_list.txt
│   ├── val_list.txt
│   └── ...
├── scripts/
│   ├── train.sh
│   ├── test.sh
│   └── ...
├── src/
│   ├── model.py
│   ├── dataset.py
│   └── ...
├── utils/
│   ├── utils.py
│   └── ...
├── LICENSE
├── README.md
└── requirements.txt

目录结构介绍

• config_files: 存放项目的配置文件，如训练和测试的参数配置。
• doc: 存放项目的文档文件，如README.md。
• lists: 存放数据集的列表文件，如训练集和验证集的列表。
• scripts: 存放项目的脚本文件，如训练和测试的脚本。
• src: 存放项目的主要源代码文件，如模型定义和数据集处理。
• utils: 存放项目的工具函数文件。
• LICENSE: 项目的开源许可证文件。
• README.md: 项目的介绍和使用说明文件。
• requirements.txt: 项目的依赖库文件。

2. 项目的启动文件介绍

训练脚本

训练脚本位于 scripts/train.sh，用于启动训练过程。使用方法如下：

sh scripts/train.sh [data] [split] [[gpu_ids]] [layers] [shots]

• [data]: 数据集名称，如 pascal。
• [split]: 数据集的分割编号。
• [gpu_ids]: GPU设备编号，如 0。
• [layers]: 使用的模型层数，如 50。
• [shots]: 使用的样本数量，如 1。

测试脚本

测试脚本位于 scripts/test.sh，用于启动测试过程。使用方法如下：

sh scripts/test.sh [data] [shot] [[gpu_ids]] [layers] [split]

• [data]: 数据集名称，如 pascal。
• [shot]: 使用的样本数量，如 1。
• [gpu_ids]: GPU设备编号，如 0。
• [layers]: 使用的模型层数，如 50。
• [split]: 数据集的分割编号。

3. 项目的配置文件介绍

配置文件位于 config_files/ 目录下，常见的配置文件包括 config1.yaml 和 config2.yaml。配置文件中包含了训练和测试的各种参数设置，如数据集路径、模型参数、优化器参数等。

配置文件示例

data_root: "/path/to/dataset"
train_list: "lists/train_list.txt"
val_list: "lists/val_list.txt"
resume_weights: "/path/to/pretrained/weights"
batch_size: 8
learning_rate: 0.001
num_epochs: 50

配置文件参数说明

• data_root: 数据集的根目录路径。
• train_list: 训练集的列表文件路径。
• val_list: 验证集的列表文件路径。
• resume_weights: 预训练模型的权重文件路径。
• batch_size: 批处理大小。
• learning_rate: 学习率。
• num_epochs: 训练的总轮数。

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