franciszzj / SGCond

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SGCond

Generating images from scene graphs using conditional control of diffusion models.

Header

Info

  • Right now we use a modified version of sg2im for scene layout generation
  • Scene graphs are stored in data.json
  • Scene layout generation (bounding boxes and semantic maps) is done in prepare.ipynb
  • ControlNet.ipynb creates images from semantic maps
  • GLIGEN.ipynb creates images from bounding boxes
  • evaluate.ipynb measures insception score and FID score

About


Languages

Language:Jupyter Notebook 99.1%Language:Python 0.9%Language:Shell 0.0%