CDVD-TSP 项目使用教程

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

CDVD-TSP 项目的目录结构如下：

CDVD-TSP/
├── dataset/
├── infer_results/
├── pretrain_models/
├── code/
│   ├── dataset/
│   ├── infer_results/
│   ├── pretrain_models/
│   ├── gitignore
│   ├── LICENSE
│   ├── README.md
│   └── ...
├── gitignore
├── LICENSE
└── README.md

目录介绍

• dataset/: 存放训练和测试数据集的目录。
• infer_results/: 存放推理结果的目录。
• pretrain_models/: 存放预训练模型的目录。
• code/: 项目的主要代码目录，包含数据处理、模型训练和推理等代码。
• gitignore: Git 忽略文件配置。
• LICENSE: 项目许可证文件。
• README.md: 项目说明文档。

2. 项目的启动文件介绍

项目的启动文件主要位于 code/ 目录下，具体包括：

• train.py: 用于训练模型的启动文件。
• test.py: 用于测试模型的启动文件。
• inference.py: 用于推理的启动文件。

启动文件介绍

• train.py: 该文件包含模型训练的逻辑，可以通过命令行参数配置训练参数。
• test.py: 该文件用于测试训练好的模型，评估模型性能。
• inference.py: 该文件用于对新数据进行推理，生成去模糊后的视频。

3. 项目的配置文件介绍

项目的配置文件主要位于 code/ 目录下，具体包括：

• config.yaml: 包含项目的主要配置参数，如数据路径、模型参数、训练参数等。

配置文件介绍

• config.yaml: 该文件采用 YAML 格式，包含以下主要配置项：
  • data_path: 数据集路径。
  • model_params: 模型参数，如卷积层数、滤波器大小等。
  • train_params: 训练参数，如学习率、批次大小、训练轮数等。
  • test_params: 测试参数，如测试数据集路径、评估指标等。

通过修改 config.yaml 文件，可以灵活调整项目的运行配置。

以上是 CDVD-TSP 项目的基本使用教程，希望对您有所帮助。