UDAT 项目使用教程

UDAT 项目使用教程

UDAT 项目地址: https://gitcode.com/gh_mirrors/ud/UDAT

1. 项目目录结构及介绍

UDAT 项目的目录结构如下：

UDAT/
├── preprocessing/
│   ├── lowlight_enhancement.py
│   ├── inference.py
│   ├── gen_seq_bboxes.py
│   ├── par_crop.py
│   └── gen_json.py
├── UDAT/
│   ├── CAR/
│   │   ├── experiments/
│   │   └── tools/
│   │       ├── train.py
│   │       ├── test.py
│   │       └── eval.py
│   └── tools/
│       └── snapshot/
├── LICENSE
└── README.md

目录介绍

• preprocessing/: 包含预处理脚本，用于生成训练数据。

  • lowlight_enhancement.py: 用于增强夜间序列的脚本。
  • inference.py: 用于预测显著对象并获取候选框的脚本。
  • gen_seq_bboxes.py: 用于从候选框生成伪注释的脚本。
  • par_crop.py: 用于生成裁剪训练补丁的脚本。
  • gen_json.py: 用于生成训练 JSON 文件的脚本。

• UDAT/: 包含 UDAT 框架的核心代码。

  • CAR/: 包含 UDAT-CAR 模型的实验和工具。
    • experiments/: 存放实验结果和模型文件。
    • tools/: 包含训练、测试和评估脚本。
      • train.py: 用于训练模型的脚本。
      • test.py: 用于测试模型的脚本。
      • eval.py: 用于评估模型的脚本。
  • tools/: 包含预训练模型和其他工具。
    • snapshot/: 存放预训练模型的目录。

• LICENSE: 项目的许可证文件。

• README.md: 项目的介绍和使用说明。

2. 项目启动文件介绍

训练启动文件

训练模型的启动文件是 UDAT/CAR/tools/train.py。使用以下命令启动训练：

cd UDAT/CAR
export PYTHONPATH=$PWD
python tools/train.py

测试启动文件

测试模型的启动文件是 UDAT/CAR/tools/test.py。使用以下命令启动测试：

python tools/test.py --dataset NAT

评估启动文件

评估模型的启动文件是 UDAT/CAR/tools/eval.py。使用以下命令启动评估：

python tools/eval.py --dataset NAT

3. 项目配置文件介绍

UDAT 项目没有显式的配置文件，但可以通过命令行参数或环境变量进行配置。主要的配置项包括：

• 数据集路径: 在 preprocessing/lowlight_enhancement.py 中配置训练数据集的路径。
• 预训练模型路径: 在 UDAT/tools/snapshot/ 中存放预训练模型。
• 实验路径: 在 UDAT/CAR/experiments/ 中存放实验结果和模型文件。

通过这些配置，用户可以根据自己的需求调整数据集和模型的路径，以适应不同的训练和测试环境。

UDAT 项目地址: https://gitcode.com/gh_mirrors/ud/UDAT