LayoutGPT 项目使用教程

1. 项目目录结构及介绍

LayoutGPT 项目的目录结构如下：

LayoutGPT/
├── ATISS/
│   ├── assets/
│   ├── dataset/
│   ├── eval_models/
│   │   └── GLIP/
│   ├── gligen/
│   ├── llm_output/
│   ├── gitignore
│   ├── LICENSE
│   ├── README.md
│   ├── environment.yml
│   ├── eval_counting_layout.py
│   ├── eval_scene_layout.py
│   ├── eval_spatial_layout.py
│   ├── parse_llm_output.py
│   ├── requirements.txt
│   ├── run_layoutgpt_2d.py
│   ├── run_layoutgpt_3d.py
│   └── utils.py
└── ...

目录结构介绍

• ATISS/: 包含与 ATISS 相关的文件和脚本。
  • assets/: 存放项目所需的静态资源文件。
  • dataset/: 存放数据集文件，包括预处理后的数据。
  • eval_models/: 包含用于评估的模型文件，如 GLIP。
  • gligen/: 包含与 GLIGEN 相关的文件和脚本。
  • llm_output/: 存放生成的布局输出文件。
  • gitignore: Git 忽略文件配置。
  • LICENSE: 项目许可证文件。
  • README.md: 项目说明文档。
  • environment.yml: Conda 环境配置文件。
  • eval_counting_layout.py: 用于评估计数布局的脚本。
  • eval_scene_layout.py: 用于评估场景布局的脚本。
  • eval_spatial_layout.py: 用于评估空间布局的脚本。
  • parse_llm_output.py: 用于解析 LLM 输出的脚本。
  • requirements.txt: 项目依赖库列表。
  • run_layoutgpt_2d.py: 用于生成 2D 布局的启动脚本。
  • run_layoutgpt_3d.py: 用于生成 3D 布局的启动脚本。
  • utils.py: 包含项目中使用的实用工具函数。

2. 项目启动文件介绍

run_layoutgpt_2d.py

该脚本用于生成 2D 图像布局。启动该脚本时，可以通过命令行参数指定不同的配置选项，例如：

python run_layoutgpt_2d.py --icl_type k-similar --K 8 --setting counting --llm_type gpt4 --n_iter 5

run_layoutgpt_3d.py

该脚本用于生成 3D 室内场景布局。启动该脚本时，可以通过命令行参数指定不同的配置选项，例如：

python run_layoutgpt_3d.py --dataset_dir /ATISS/data_output --icl_type k-similar --K 8 --room bedroom --llm_type gpt4 --unit px --normalize --regular_floor_plan

3. 项目的配置文件介绍

environment.yml

该文件用于配置 Conda 环境，包含了项目所需的所有依赖库。可以通过以下命令创建并激活环境：

conda env create -f environment.yml
conda activate layoutgpt

requirements.txt

该文件列出了项目所需的所有 Python 依赖库。可以通过以下命令安装这些依赖：

pip install -r requirements.txt

其他配置文件

• gitignore: 配置 Git 忽略的文件和目录。
• LICENSE: 项目许可证文件，定义了项目的使用条款。
• README.md: 项目说明文档，包含了项目的概述、安装步骤、使用方法等信息。