Skip to content

design-generators

design-generators ports layout, poster, and graphic-design generation research repositories into framework-specific packages for 🤗transformers, 🧨diffusers, and 🤖pydantic-ai that can load converted weights or prompt configuration and run inference through a consistent public schema.

Highlights

Models

The model name opens the on-site API page, and Docs links to the package README. Weights are converted original weights you can load, while Training means this repository trained and verified the weights itself.

Model Task Venue Weights Training Paper Docs
BASNet Saliency detection venue: CVPR 2019 checkpoint: ckpt training: n/a Paper README
CGB-DM Poster layout generation venue: arXiv 2024 checkpoint: ckpt training: train Paper README
Coarse-to-Fine Layout generation venue: AAAI 2022 checkpoint: ckpt training: n/a Paper README
DLT Layout generation venue: ICCV 2023 checkpoint: ckpt training: train Paper README
DS-GAN Poster layout generation venue: CVPR 2023 checkpoint: ckpt training: n/a Paper README
Flex-DM Layout generation venue: CVPR 2023 checkpoint: ckpt training: n/a Paper README
House-GAN Floorplan generation venue: ECCV 2020 checkpoint: ckpt training: n/a Paper README
LACE Layout generation venue: ICLR 2024 checkpoint: ckpt training: n/a Paper README
LayouSyn Layout generation venue: ICCV 2025 checkpoint: ckpt training: n/a Paper README
Layout-Corrector Layout generation venue: ECCV 2024 checkpoint: ckpt training: n/a Paper README
LayoutDETR Content-image layout generation venue: ECCV 2024 checkpoint: ckpt training: n/a Paper README
LayoutDM Layout generation venue: CVPR 2023 checkpoint: ckpt training: train Paper README
Layout FID Layout evaluation venue: ACM MM 2021 checkpoint: ckpt training: n/a Paper README
LayoutAction Layout generation venue: AAAI 2023 checkpoint: ckpt training: n/a Paper README
LayoutFlow Layout generation venue: ECCV 2024 checkpoint: ckpt training: train Paper README
LayoutGPT Prompt-based layout generation venue: NeurIPS 2023 checkpoint: ckpt training: n/a Paper README
LT-Net Scene-graph-to-layout generation venue: CVPR 2021 checkpoint: ckpt training: n/a Paper README
LayoutDiffusion Layout generation venue: ICCV 2023 checkpoint: ckpt training: train Paper README
LayoutFormer++ Layout generation venue: CVPR 2023 checkpoint: ckpt training: n/a Paper README
LayoutGAN++ Layout generation venue: ACM MM 2021 checkpoint: ckpt training: n/a Paper README
LayoutVAE Layout generation venue: ICCV 2019 checkpoint: ckpt training: n/a Paper README
LayoutPrompter Prompt-based layout generation venue: NeurIPS 2023 checkpoint: ckpt training: n/a Paper README
Parse-Then-Place Layout generation venue: ICCV 2023 checkpoint: ckpt training: n/a Paper README
PosterLlama Poster layout generation venue: ECCV 2024 checkpoint: ckpt training: n/a Paper README
PosterLLaVA Poster layout generation venue: TMM 2024 checkpoint: ckpt training: n/a Paper README
PosterO Poster layout generation venue: CVPR 2025 checkpoint: ckpt training: n/a Paper README
RALF Retrieval-augmented layout generation venue: CVPR 2024 checkpoint: ckpt training: n/a Paper README
SmartText Text placement venue: TMM 2021 checkpoint: ckpt training: n/a Paper README

Libraries

Library Description Docs
library: laygen Layout-generation schemas, pipeline helpers, bbox utilities, schedulers, model-card helpers, and testing helpers. README
library: posgen Poster-generation and content-aware placement contracts for shared dataset names, position content, and label helpers. README
library: traingen Training utilities for package-local PyTorch Lightning CLI integration in packages that train models locally. README
library: traingen-parity Deterministic trace capture and comparison helpers for training-parity checks. README

Quick Start

Install the shared layout library directly from this repository:

pip install "laygen @ git+https://github.com/creative-graphic-design/design-generators.git#subdirectory=lib/laygen"

Model packages depend on workspace libraries that are not published on PyPI, so install laygen together with the model package:

pip install \
  "laygen @ git+https://github.com/creative-graphic-design/design-generators.git#subdirectory=lib/laygen" \
  "layout-dm @ git+https://github.com/creative-graphic-design/design-generators.git#subdirectory=models/layout-dm"
from layout_dm import LayoutDMPipeline

print(LayoutDMPipeline.__name__)

For development, clone the repository and run member commands from the repository root with uv run --package <member> ...; each package documents its reproduction workflow in REPRODUCING.md.

git clone https://github.com/creative-graphic-design/design-generators.git
cd design-generators
uv run --package layout-dm pytest models/layout-dm/tests -m "not vendor_parity and not integration"

Getting started

The Getting Started guide explains the complete installation and first-inference workflow.

How the repository is organized

The architecture guide explains the boundary between shared libraries and model packages, while Conventions defines the common public output schema and conditioning names.

Policies

Reproducibility

Every port is verified against the original implementation: agreement checks (the repository's vendor_parity tests) compare package outputs with references generated by running the original implementation in vendor/. Each package's REPRODUCING.md shows how to reproduce the check, and its README reports the resulting comparison numbers.

Training

For models we reproduce by training ourselves, each package provides PyTorch Lightning modules, LightningCLI configs, and package-specific training commands. Shared training helpers live in lib/traingen, parity helpers live in lib/traingen-parity, and runnable workflows are documented in each package's TRAINING.md; see the training reproduction guide for the repository protocol.

Extension

Extend by installing pinned released packages and subclassing; never copy framework or repository source trees into your project and mutate them in place. This applies to human developers and AI coding agents alike; see Extending.

API reference

See the design-generators documentation for guides and the API reference for workspace libraries and model packages.

License

Repository code is licensed under Apache-2.0; see LICENSE. Converted weights, datasets, and vendored upstream code carry their original licenses.