README.md

---
library_name: pytorch
license: mit
tags:
  - computer-vision
  - floorplan-reconstruction
  - pytorch
  - raster2seq
---

<div align="center">
  <h1>Official PyTorch models of "Raster2Seq: Polygon Sequence Generation for Floorplan Reconstruction" <a href="https://arxiv.org/abs/2602.09016"> (SIGGRAPH'26)</a></h1>
</div>

<div align="center">
  <a href="https://hao-pt.github.io/" target="_blank">Hao&nbsp;Phung</a> &emsp; &emsp;
  <a href="https://hao-pt.github.io/" target="_blank">Hadar&nbsp;Averbuch-Elor</a>
  <br> <br>
  Cornell University &emsp;
  <br> <br>
  <a href="https://cornell-vailab.github.io/Raster2Seq/">[Page]</a> &emsp;&emsp;
  <a href="https://arxiv.org/abs/2602.09016">[Paper]</a> &emsp;&emsp;
  <a href="https://github.com/Cornell-VAILab/Raster2Seq">[Code]</a> &emsp;&emsp;
  <br> <br>
</div>

This repository hosts Raster2Seq's PyTorch checkpoints for floorplan reconstruction. Each checkpoint is stored in its own subfolder so users can download all checkpoints or only the subfolder they need.

## Available Checkpoints

| Checkpoint key     | Dataset                 | RoomF1 | Subfolder           |
| ------------------ | ----------------------- | -----: | ------------------- |
| `s3d-bw`           | Structured3D-B          |   99.6 | `s3d-bw/`           |
| `cubicasa5k`       | CubiCasa5K              |   88.7 | `cubicasa5k/`       |
| `raster2graph`     | Raster2Graph            |   97.0 | `raster2graph/`     |
| `raster2graph-512` | Raster2Graph-512        |   98.1 | `Raster2Graph-512/` |
| `s3d-density`      | Structured3D-DensityMap |   99.1 | `s3d-density/`      |

## Download All Checkpoints

```python
from huggingface_hub import snapshot_download

local_repo = snapshot_download(repo_id="haopt/Raster2Seq")
```

## Download One Checkpoint Subfolder

```python
from huggingface_hub import snapshot_download

local_repo = snapshot_download(
    repo_id="haopt/Raster2Seq",
    allow_patterns="Raster2Graph-512/*",
)
```

## Raster2Seq Helper

With the Raster2Seq codebase, users can load by alias:

```bash
python eval.py --checkpoint hf:cubicasa5k ...
```

or download directly:

```python
from raster2seq_hub import download_checkpoint

ckpt_path = download_checkpoint("cubicasa5k")
```

**Please CITE** our paper and give us a :star: whenever this repository is used to help produce published results or incorporated into other software.

```bibtex
@inproceedings{phung2026raster2seq,
   title={Raster2Seq: Polygon Sequence Generation for Floorplan Reconstruction},
   author={Phung, Hao and Averbuch-Elor, Hadar},
   booktitle={Special Interest Group on Computer Graphics and Interactive Techniques Conference Conference Papers},
   year= {2026},
}
```