ReferringRelationships 项目使用教程

ReferringRelationships 项目使用教程

ReferringRelationships 项目地址: https://gitcode.com/gh_mirrors/re/ReferringRelationships

1. 项目目录结构及介绍

ReferringRelationships 项目是一个用于处理和识别图像中物体间关系的研究项目。项目目录结构如下：

ReferringRelationships/
├── scripts/             # 脚本目录，包含数据下载和预处理脚本
├── data.py              # 数据处理脚本，用于生成训练和测试数据
├── evaluate.py          # 评估脚本，用于评估模型性能
├── evaluate_discovery.py # 发现关系评估脚本
├── evaluate_per_relationship.ipynb # 单个关系评估脚本
├── evaluation.ipynb     # 评估流程脚本
├── get_best_model.py    # 获取最佳模型脚本
├── iterator.py          # 数据迭代器脚本
├── launch.py            # 启动训练脚本
├── models.py            # 模型定义脚本
├── requirements.txt     # 项目依赖
├── resnet_helpers.py    # ResNet 辅助函数脚本
├── train.py             # 训练模型脚本
├── visualization.ipynb  # 可视化结果脚本
├── attention_saccades.ipynb # 注意力模型分析脚本
├── controlled_shift.ipynb # 控制位移分析脚本
├── __init__.py          # 初始化文件
├── _config.yml          # 配置文件
├── config.py            # 配置配置文件
└── README.md            # 项目说明文件

2. 项目的启动文件介绍

项目的启动主要是通过 scripts 目录下的脚本进行。以下是一些主要的启动脚本：

• download_data.sh: 用于下载项目所需的数据集。
• create_vrd_dataset.sh: 用于创建 VRD 数据集。
• train_vrd.sh: 用于训练 VRD 数据集的模型。
• evaluate_vrd.sh: 用于评估 VRD 数据集的模型。
• create_clevr_dataset.sh: 用于创建 CLEVR 数据集。
• train_clevr.sh: 用于训练 CLEVR 数据集的模型。
• evaluate_clevr.sh: 用于评估 CLEVR 数据集的模型。
• create_visualgenome_dataset.sh: 用于创建 Visual Genome 数据集。
• train_visualgenome.sh: 用于训练 Visual Genome 数据集的模型。
• evaluate_visualgenome.sh: 用于评估 Visual Genome 数据集的模型。

3. 项目的配置文件介绍

项目中的配置文件主要包括 _config.yml 和 config.py。

• _config.yml: 这个文件包含了 Jekyll 的一些基础配置，例如主题设置等。
• config.py: 这个文件包含了项目运行时的配置信息，如模型参数、数据路径、训练参数等。用户可以通过修改这个文件来调整模型的训练和测试行为。

确保在开始训练和测试前正确配置这些文件，以保障项目能够按照用户的需求运行。

ReferringRelationships 项目地址: https://gitcode.com/gh_mirrors/re/ReferringRelationships