Upload folder using huggingface_hub

README.md

@@ -1,3 +1,61 @@
 ---
+library_name: pytorch
 license: mit
+tags:
+  - computer-vision
+  - floorplan-reconstruction
+  - pytorch
+  - raster2seq
 ---
+
+# Raster2Seq Checkpoints
+
+This repository hosts Raster2Seq PyTorch checkpoints for polygon sequence generation from rasterized floorplans. 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/` |
+| `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="cubicasa5k/*",
+)
+```
+
+## 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")
+```
+
+The helper downloads the matching subfolder and returns the cached `checkpoint.pth` path.
+
+Hugging Face repository IDs use the form `namespace/repo`. Dataset-specific choices are represented by subfolders, not by repo IDs such as `Raster2Seq/cubicasa5k`.

cubicasa5k/checkpoint.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f7c0ef9379aa8cf11349da8b2179214e0d495af758383984a2361a1309aab3b7
+size 1452141664

cubicasa5k/config.json

@@ -0,0 +1,26 @@
+{
+  "checkpoint_key": "cubicasa5k",
+  "name": "CubiCasa5K",
+  "checkpoint_filename": "checkpoint.pth",
+  "source_local_filename": "cc5k_sem_res256_ep0499.pth",
+  "metrics": {
+    "room_f1": 88.7
+  },
+  "inference_args": {
+    "dataset_name": "cubicasa",
+    "dataset_root": "data/coco_cubicasa5k_nowalls_v4-1_refined/",
+    "eval_set": "test",
+    "output_dir": "eval_outputs/cc5k_sem_results/",
+    "semantic_classes": 12,
+    "input_channels": 3,
+    "poly2seq": true,
+    "seq_len": 512,
+    "num_bins": 32,
+    "disable_poly_refine": true,
+    "dec_attn_concat_src": true,
+    "ema4eval": true,
+    "use_anchor": true,
+    "per_token_sem_loss": true,
+    "save_pred": true
+  }
+}

manifest.json

@@ -0,0 +1,45 @@
+{
+  "repo_id": "haopt/Raster2Seq",
+  "checkpoint_filename": "checkpoint.pth",
+  "checkpoints": {
+    "s3d-bw": {
+      "name": "Structured3D-B",
+      "subfolder": "s3d-bw",
+      "local_filename": "s3dbw_sem_res256_ep0449.pth",
+      "room_f1": 99.6,
+      "config": "s3d-bw/config.json"
+    },
+    "cubicasa5k": {
+      "name": "CubiCasa5K",
+      "subfolder": "cubicasa5k",
+      "local_filename": "cc5k_sem_res256_ep0499.pth",
+      "room_f1": 88.7,
+      "config": "cubicasa5k/config.json"
+    },
+    "raster2graph": {
+      "name": "Raster2Graph",
+      "subfolder": "raster2graph",
+      "local_filename": "r2g_sem_res256_ep0549.pth",
+      "room_f1": 97.0,
+      "config": "raster2graph/config.json"
+    },
+    "s3d-density": {
+      "name": "Structured3D-DensityMap",
+      "subfolder": "s3d-density",
+      "local_filename": "s3dd_sem_res256_ep0699.pth",
+      "room_f1": 99.1,
+      "config": "s3d-density/config.json"
+    }
+  },
+  "aliases": {
+    "s3d": "s3d-bw",
+    "structured3d": "s3d-bw",
+    "structured3d-b": "s3d-bw",
+    "s3dbw": "s3d-bw",
+    "cc5k": "cubicasa5k",
+    "cubicasa": "cubicasa5k",
+    "r2g": "raster2graph",
+    "s3dd": "s3d-density",
+    "structured3d-density": "s3d-density"
+  }
+}

raster2graph/checkpoint.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ac0ebb526a433ca2a4391362ee602869161dc052a4ff45b09d25dc7b85105610
+size 1027373728

raster2graph/config.json

@@ -0,0 +1,26 @@
+{
+  "checkpoint_key": "raster2graph",
+  "name": "Raster2Graph",
+  "checkpoint_filename": "checkpoint.pth",
+  "source_local_filename": "r2g_sem_res256_ep0549.pth",
+  "metrics": {
+    "room_f1": 97.0
+  },
+  "inference_args": {
+    "dataset_name": "r2g",
+    "dataset_root": "data/R2G_hr_dataset_processed_v1",
+    "eval_set": "test",
+    "output_dir": "eval_outputs/r2g_sem_results",
+    "semantic_classes": 13,
+    "input_channels": 3,
+    "poly2seq": true,
+    "seq_len": 512,
+    "num_bins": 32,
+    "disable_poly_refine": true,
+    "dec_attn_concat_src": true,
+    "ema4eval": true,
+    "use_anchor": true,
+    "per_token_sem_loss": true,
+    "save_pred": true
+  }
+}

s3d-bw/checkpoint.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f5b82e15e033f1394dc6bfb7466396da30f04a76cbb1c1ec44024c1e60702fd0
+size 983456736

s3d-bw/config.json

@@ -0,0 +1,26 @@
+{
+  "checkpoint_key": "s3d-bw",
+  "name": "Structured3D-B",
+  "checkpoint_filename": "checkpoint.pth",
+  "source_local_filename": "s3dbw_sem_res256_ep0449.pth",
+  "metrics": {
+    "room_f1": 99.6
+  },
+  "inference_args": {
+    "dataset_name": "stru3d",
+    "dataset_root": "data/coco_s3d_bw",
+    "eval_set": "test",
+    "output_dir": "eval_outputs/s3dbw_sem_results",
+    "semantic_classes": 19,
+    "input_channels": 3,
+    "poly2seq": true,
+    "seq_len": 512,
+    "num_bins": 32,
+    "disable_poly_refine": true,
+    "dec_attn_concat_src": true,
+    "ema4eval": true,
+    "use_anchor": true,
+    "per_token_sem_loss": true,
+    "save_pred": true
+  }
+}

s3d-density/checkpoint.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:35c7f69b8142ec26734a9f69645b5b32235030ed5e577951bc19f16c6c65b69b
+size 983406560

s3d-density/config.json

@@ -0,0 +1,26 @@
+{
+  "checkpoint_key": "s3d-density",
+  "name": "Structured3D-DensityMap",
+  "checkpoint_filename": "checkpoint.pth",
+  "source_local_filename": "s3dd_sem_res256_ep0699.pth",
+  "metrics": {
+    "room_f1": 99.1
+  },
+  "inference_args": {
+    "dataset_name": "stru3d",
+    "dataset_root": "data/stru3d",
+    "eval_set": "test",
+    "output_dir": "eval_outputs/s3dd_sem_results",
+    "semantic_classes": 19,
+    "input_channels": 1,
+    "poly2seq": true,
+    "seq_len": 512,
+    "num_bins": 32,
+    "disable_poly_refine": true,
+    "dec_attn_concat_src": true,
+    "ema4eval": true,
+    "use_anchor": true,
+    "per_token_sem_loss": true,
+    "save_pred": true
+  }
+}