CyberAgentAILab / layout-dm

LayoutDM: Discrete Diffusion Model for Controllable Layout Generation [Inoue+, CVPR2023]
https://cyberagentailab.github.io/layout-dm
Apache License 2.0
225 stars 24 forks source link

customize condition #21

Closed kietna1809 closed 1 year ago

kietna1809 commented 1 year ago

Do you have any guide to generate new layout based on customize condition. I am trying generate layout by parsing my own condition to LayoutDM but don't know where to start. As far as my research the condition fit to model are been extracted from Publaynet and Rico dataset. Another question is "Can the model generate layout by using combination of multiple condition, for instance generate layout using both relation + cwh condition" Cheer thanks

naoto0804 commented 1 year ago

Do you have any guide to generate new layout based on customize condition

We've defined some common use-cases as "task" here. If these cases do not meet your requirements, then you can define your task and implement get_cond for that task by hiding all the fields that you want to predict.

using both relation + cwh condition

It's possible. Relation information is enforced here.