Modeling Image Composition for Complex Scene Generation

By Zuopeng Yang, Daqing Liu, Chaoyue Wang, Jie Yang, Dacheng Tao. 2022

64 citations recorded by the source corpus.

We present a method that achieves state-of-the-art results on challenging (few-shot) layout-to-image generation tasks by accurately modeling textures, structures and relationships contained in a complex scene. After compressing RGB images into patch tokens, we propose the Transformer with Focal Attention (TwFA) for exploring dependencies of object-to-object, object-to-patch and patch-to-patch. Compared to existing CNN-based and Transformer-based generation models that entangled modeling on pixel-level&patch-level and object-level&patch-level respectively, the proposed focal attention predicts the current patch token by only focusing on its highly-related tokens that specified by the spatial layout, thereby achieving disambiguation during training. Furthermore, the proposed TwFA largely increases the data efficiency during training, therefore we propose the first few-shot complex scene generation strategy based on the well-trained TwFA. Comprehensive experiments show the superiority of our method, which significantly increases both quantitative metrics and qualitative visual realism with respect to state-of-the-art CNN-based and transformer-based methods. Code is available at https://github.com/JohnDreamer/TwFA.

Subfields: Computer Graphics and Computer-Aided Design, Computer Vision and Pattern Recognition, Generative Models

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computer Graphics and Computer-Aided Design

  1. Lu Yuan
  2. Andrea Vedaldi
  3. Mike Zheng Shou
  4. Varun Jampani
  5. X. Wang
  6. Sugato Basu
  7. Dongdong Chen
  8. Xun Huang

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Ming-Hsuan Yang
  4. Andrea Vedaldi
  5. Lu Yuan
  6. Xiaogang Wang
  7. Andrew Zisserman
  8. Cordelia Schmid

In Generative Models

  1. Lu Yuan
  2. Dongdong Chen
  3. Yu Cheng
  4. Minghao Chen
  5. William Yang Wang
  6. Andrea Vedaldi
  7. Mike Zheng Shou
  8. Yu-Chiang Frank Wang

Citations in the corpus