Visual Prompt Retrieval 项目使用教程

1. 项目目录结构及介绍

visual_prompt_retrieval/
├── evaluate/
│   ├── evaluate.py
│   └── ...
├── evaluate_detection/
│   ├── evaluate_detection.py
│   └── ...
├── figures/
│   ├── figures.py
│   └── ...
├── figures_dataset/
│   ├── figures_dataset.py
│   └── ...
├── tools/
│   ├── tools.py
│   └── ...
├── util/
│   ├── util.py
│   └── ...
├── .gitignore
├── LICENSE
├── README.md
├── SupPR.md
├── UnsupPR.md
├── __init__.py
├── engine_finetune.py
├── engine_pretrain.py
├── main_pretrain.py
├── model.yaml
├── models_mae.py
├── models_vit.py
├── requirements.txt
├── tta.py
├── viz_utils.py
└── vqgan.py

目录结构介绍

• evaluate/: 包含评估相关脚本。
• evaluate_detection/: 包含检测评估相关脚本。
• figures/: 包含生成图表的脚本。
• figures_dataset/: 包含数据集图表生成的脚本。
• tools/: 包含各种工具脚本。
• util/: 包含实用工具脚本。
• .gitignore: Git忽略文件配置。
• LICENSE: 项目许可证文件。
• README.md: 项目介绍和使用说明。
• SupPR.md: 监督提示检索的说明文档。
• UnsupPR.md: 无监督提示检索的说明文档。
• __init__.py: Python包初始化文件。
• engine_finetune.py: 微调引擎脚本。
• engine_pretrain.py: 预训练引擎脚本。
• main_pretrain.py: 主预训练脚本。
• model.yaml: 模型配置文件。
• models_mae.py: MAE模型脚本。
• models_vit.py: ViT模型脚本。
• requirements.txt: 项目依赖文件。
• tta.py: 测试时增强脚本。
• viz_utils.py: 可视化工具脚本。
• vqgan.py: VQGAN模型脚本。

2. 项目启动文件介绍

main_pretrain.py

main_pretrain.py 是项目的主启动文件，用于启动预训练过程。该文件包含了模型的初始化、数据加载、训练循环等核心逻辑。

使用方法

python main_pretrain.py --config model.yaml

engine_finetune.py

engine_finetune.py 是用于微调模型的启动文件。该文件包含了微调过程的配置和执行逻辑。

使用方法

python engine_finetune.py --config model.yaml

3. 项目配置文件介绍

model.yaml

model.yaml 是项目的配置文件，包含了模型的各种参数配置，如数据路径、模型超参数、训练参数等。

配置文件示例

data:
  path: "path/to/dataset"
  batch_size: 32

model:
  name: "vit"
  learning_rate: 0.001

train:
  epochs: 100
  save_interval: 10

requirements.txt

requirements.txt 是项目的依赖文件，列出了项目运行所需的所有Python包及其版本。

安装依赖

pip install -r requirements.txt

通过以上步骤，您可以顺利启动并配置 visual_prompt_retrieval 项目。