Rakutenipoi/Indoor-Layout-Generation

Indoor layout generation

Quick start

Python 3.9

Pytorch 2.1.0

# pytorch
## conda install
conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia
## or pip install
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121

Other modules

pip install tqdm pyyaml wandb 

Dataset

dataset.pkl

Place in dataset folder