Skip to content

A PyTorch implementation of the paper: Specifying Object Attributes and Relations in Interactive Scene Generation

Notifications You must be signed in to change notification settings

demo3d/scene_generation

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Specifying Object Attributes and Relations in Interactive Scene Generation

A PyTorch implementation of Specifying Object Attributes and Relations in Interactive Scene Generation

Paper

Specifying Object Attributes and Relations in Interactive Scene Generation
Oron Ashual1, Lior Wolf1,2
1 Tel-Aviv University, 2 Facebook AI Research
IEEE International Conference on Computer Vision (ICCV), 2019, (Oral)

Network Architechture

Youtube

Specifying Object Attributes and Relations in Interactive Scene Generation

Usage

1. Creating virtual environment (optional)

All code was developed and tested on Ubuntu 18.04 with Python 3.6 (Anaconda) and PyTorch 1.0.

$ conda create -n scene_generation python=3.6
$ conda activate scene_generation

2. Install COCO API

$ cd ~
$ git clone https://github.com/cocodataset/cocoapi.git
$ cd cocoapi/PythonAPI/
$ python setup.py install
$ cd ..

3. Cloning the repository

$ git clone git@github.com:ashual/scene_generation.git
$ cd scene_generation

4. Installing dependencies

$ conda install -r requirements.txt

5. Training

$ python train.py

6. Downloading trained models

TBD

7. GUI

The GUI was built as POC. For using it run:

python scripts/gui/simple-server.py --checkpoint YOUR_MODEL_CHECKPOINT

Citation

If you find this code useful in your research then please cite

@inproceedings{ashual2019scenegeneration,
  title={Specifying Object Attributes and Relations in Interactive Scene Generation},
  author={Ashual, Oron and Wolf, Lior},
  booktitle={ICCV},
  year={2019}
}

Acknowledgement

Our project borrows some source files from sg2im. We thank the authors.

About

A PyTorch implementation of the paper: Specifying Object Attributes and Relations in Interactive Scene Generation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 86.6%
  • HTML 6.6%
  • JavaScript 4.2%
  • CSS 2.3%
  • Shell 0.3%