中文

将 HouseDiffusion 适配至 Modified Swiss Dwellings 数据集以进行条件化户型生成

计算机视觉与模式识别 2023-12-08 v1

摘要

随着若干方法的提出,自动户型生成近期获得了快速发展。CVAAD Floor Plan Auto-Completion 研讨会挑战赛引入了 MSD,这是一个新数据集,包含建筑现有结构墙作为额外的输入约束。本技术报告提出了一种将近期工作 HouseDiffusion (arXiv:2211.13287 [cs.CV]) 扩展至 MSD 数据集的方法。该适配涉及修改模型的 Transformer 层,使其以一组墙线为条件。报告引入了一个预处理流水线,用于从作为输入提供的建筑结构二值掩码中提取墙线。此外,研究发现将所有房间多边形简化为矩形的数据处理过程能带来更好的性能。这表明未来的工作应探索扩散模型中变长多边形的更好表示。代码将在日后发布。

关键词

引用

@article{arxiv.2312.03938,
  title  = {Adapting HouseDiffusion for conditional Floor Plan generation on Modified Swiss Dwellings dataset},
  author = {Emanuel Kuhn},
  journal= {arXiv preprint arXiv:2312.03938},
  year   = {2023}
}