University1652-Baseline 项目使用教程

1. 项目目录结构及介绍

University1652-Baseline/
├── README.md
├── train/
│   ├── drone/ /* 无人机视角训练图像
│   ├── street/ /* 街道视角训练图像
│   ├── satellite/ /* 卫星视角训练图像
│   ├── google/ /* 从Google Image收集的噪声街道视角训练图像
├── test/
│   ├── query_drone/
│   ├── gallery_drone/
│   ├── query_street/
│   ├── gallery_street/
│   ├── query_satellite/
│   ├── gallery_satellite/
├── 4K_drone/
├── train.py
├── test.py
├── config.py
├── requirements.txt
├── utils/
│   ├── __init__.py
│   ├── ...
├── models/
│   ├── __init__.py
│   ├── ...
├── data_loader/
│   ├── __init__.py
│   ├── ...
└── ...

目录结构说明

• train/: 包含训练数据集，分为无人机视角、街道视角和卫星视角。
• test/: 包含测试数据集，分为无人机视角、街道视角和卫星视角的查询和图库图像。
• 4K_drone/: 包含4K分辨率的无人机图像。
• train.py: 训练脚本。
• test.py: 测试脚本。
• config.py: 配置文件。
• requirements.txt: 项目依赖文件。
• utils/: 包含各种实用工具函数。
• models/: 包含模型定义文件。
• data_loader/: 包含数据加载器。

2. 项目启动文件介绍

train.py

train.py 是项目的训练脚本，用于训练模型。主要功能包括：

• 加载数据集
• 定义模型
• 定义损失函数
• 训练模型
• 保存模型

test.py

test.py 是项目的测试脚本，用于评估模型的性能。主要功能包括：

• 加载测试数据集
• 加载训练好的模型
• 进行预测
• 计算评估指标

3. 项目的配置文件介绍

config.py

config.py 是项目的配置文件，包含了项目的各种配置参数。主要配置项包括：

• 数据集路径: 指定训练和测试数据集的路径。
• 模型参数: 定义模型的超参数，如学习率、批量大小等。
• 训练参数: 定义训练过程中的参数，如训练轮数、优化器等。
• 测试参数: 定义测试过程中的参数，如评估指标等。

示例配置

# config.py

# 数据集路径
TRAIN_DATA_PATH = 'path/to/train/data'
TEST_DATA_PATH = 'path/to/test/data'

# 模型参数
LEARNING_RATE = 0.001
BATCH_SIZE = 32

# 训练参数
NUM_EPOCHS = 100
OPTIMIZER = 'Adam'

# 测试参数
EVAL_METRICS = ['accuracy', 'precision', 'recall']

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

---

以上是 University1652-Baseline 项目的基本使用教程，希望对你有所帮助！