AI2-THOR Rearrangement 项目使用教程

AI2-THOR Rearrangement 项目使用教程

ai2thor-rearrangement🔀 Visual Room Rearrangement项目地址:https://gitcode.com/gh_mirrors/ai/ai2thor-rearrangement

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

AI2-THOR Rearrangement 项目的目录结构如下：

ai2thor-rearrangement/
├── example.py
├── baseline_configs/
│   └── rearrange_base.py
├── one_phase/
│   └── ...
├── two_phase/
│   └── ...
├── walkthrough/
│   └── ...
└── rearrange/
    └── baseline_models.py

• example.py: 示例脚本，展示如何实例化重排任务进行训练和验证。
• baseline_configs/: 包含基础配置文件的目录。
  • rearrange_base.py: 定义挑战参数的基础配置文件（例如屏幕大小、允许的动作等）。
• one_phase/: 包含1阶段挑战轨道的基础实验配置文件。
• two_phase/: 包含2阶段挑战轨道的基础实验配置文件。
• walkthrough/: 如果需要单独训练引导阶段，包含基础实验配置文件。
• rearrange/: 包含基础模型的集合。
  • baseline_models.py: 基础模型的集合文件。

2. 项目的启动文件介绍

项目的启动文件是 example.py。这个脚本展示了如何实例化重排任务进行训练和验证。你可以通过运行以下命令来启动项目：

python example.py

3. 项目的配置文件介绍

项目的配置文件位于 baseline_configs/ 目录下，主要文件是 rearrange_base.py。这个文件定义了挑战的参数，例如屏幕大小、允许的动作等。以下是配置文件的主要内容：

# rearrange_base.py

# 定义屏幕大小
SCREEN_SIZE = 224

# 定义允许的动作
ALLOWED_ACTIONS = ['move_forward', 'turn_left', 'turn_right', 'pickup_object', 'put_object']

# 其他配置参数...

通过修改这些配置参数，你可以调整项目的运行环境和行为。

ai2thor-rearrangement🔀 Visual Room Rearrangement项目地址:https://gitcode.com/gh_mirrors/ai/ai2thor-rearrangement