add citation to readme

README.md

@@ -19,7 +19,7 @@ tags:
 - multimodal
 ---
 <div align="center">
-    <h1 align="center">The P<sup>3</sup> dataset: Pixels, Points and Polygons <br> for Multimodal Building Vectorization</h1>
+    <h1 align="center">The P<sup>3</sup> Dataset: Pixels, Points and Polygons <br> for Multimodal Building Vectorization</h1>
     <h3><align="center">Raphael Sulzer<sup>1,2</sup> &nbsp;&nbsp;&nbsp; Liuyun Duan<sup>1</sup>
     &nbsp;&nbsp;&nbsp; Nicolas Girard<sup>1</sup>&nbsp;&nbsp;&nbsp; Florent Lafarge<sup>2</sup></a></h3>
     <align="center"><sup>1</sup>LuxCarta Technology <br>  <sup>2</sup>Centre Inria d'Université Côte d'Azur
@@ -48,6 +48,7 @@ We present the P<sup>3</sup> dataset, a large-scale multimodal benchmark for bui
 - A global, multimodal dataset of aerial images, aerial LiDAR point clouds and building outline polygons, available at [huggingface.co/datasets/rsi/PixelsPointsPolygons](https://huggingface.co/datasets/rsi/PixelsPointsPolygons) 
 - A library for training and evaluating state-of-the-art deep learning methods on the dataset, available at [github.com/raphaelsulzer/PixelsPointsPolygons](https://github.com/raphaelsulzer/PixelsPointsPolygons)
 - Pretrained model weights, available at [huggingface.co/rsi/PixelsPointsPolygons](https://huggingface.co/rsi/PixelsPointsPolygons) 
+- A paper with an extensive experimental validation, available at [arxiv.org/abs/2505.15379](https://arxiv.org/abs/2505.15379)
 
 ## Dataset
 
@@ -595,7 +596,15 @@ python scripts/train.py experiment=p2p_fusion checkpoint=latest
 
 If you use our work please cite
 ```bibtex
-TODO
+@misc{sulzer2025p3datasetpixelspoints,
+      title={The P$^3$ dataset: Pixels, Points and Polygons for Multimodal Building Vectorization}, 
+      author={Raphael Sulzer and Liuyun Duan and Nicolas Girard and Florent Lafarge},
+      year={2025},
+      eprint={2505.15379},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/2505.15379}, 
+}
 ```
 
 ## Acknowledgements

ffl/224/v4_image_bs4x16/.hydra/config.yaml

@@ -234,8 +234,7 @@ evaluation:
   modes:
   - iou
   eval_file: results/metrics
-experiment.name: debug
 output_dir: ${.host.model_root}/${.experiment.model.name}/${.experiment.encoder.in_size}/${.experiment.name}
 checkpoint: best_val_iou
 num_workers: ${.run_type.num_workers}
-lidar_file: demo_data/lidar0_CH_val.copc.laz
+image_file: /user/rsulzer/home/image1860_NZ_test.tif

ffl/224/v4_image_bs4x16/.hydra/hydra.yaml

@@ -116,11 +116,11 @@ hydra:
     - host=gin
     - checkpoint=best_val_iou
     - experiment=ffl_image
-    - +lidar_file=demo_data/lidar0_CH_val.copc.laz
+    - +image_file=/user/rsulzer/home/image1860_NZ_test.tif
   job:
     name: predict_demo
     chdir: null
-    override_dirname: +lidar_file=demo_data/lidar0_CH_val.copc.laz,checkpoint=best_val_iou,experiment=ffl_image,host=gin,run_type=debug
+    override_dirname: +image_file=/user/rsulzer/home/image1860_NZ_test.tif,checkpoint=best_val_iou,experiment=ffl_image,host=gin,run_type=debug
     id: ???
     num: ???
     config_name: config

ffl/224/v4_image_bs4x16/.hydra/overrides.yaml

@@ -2,4 +2,4 @@
 - host=gin
 - checkpoint=best_val_iou
 - experiment=ffl_image
-- +lidar_file=demo_data/lidar0_CH_val.copc.laz
+- +image_file=/user/rsulzer/home/image1860_NZ_test.tif

hisup/224/v3_image_vit_cnn_bs4x12/.hydra/config.yaml

@@ -7,14 +7,14 @@ host:
   update_pbar_every: 1
   ldof_exe: /user/rsulzer/home/cpp/line-DOF-metric/build/calculate_DoF
 run_type:
-  name: debug
+  name: release
   batch_size: 16
-  train_subset: 256
-  val_subset: 32
-  test_subset: 32
-  logging: DEBUG
-  num_workers: 0
-  log_to_wandb: false
+  train_subset: null
+  val_subset: null
+  test_subset: null
+  logging: INFO
+  num_workers: 16
+  log_to_wandb: true
 dataset:
   name: PixelsPointsPolygons
   size: ${..experiment.encoder.in_size}
@@ -108,8 +108,7 @@ evaluation:
   modes:
   - iou
   eval_file: results/metrics
-experiment.name: debug
 output_dir: ${.host.model_root}/${.experiment.model.name}/${.experiment.encoder.in_size}/${.experiment.name}
 checkpoint: best_val_iou
 num_workers: ${.run_type.num_workers}
-image_file: demo_data/image0_CH_val.tif
+image_file: /user/rsulzer/home/inria4.tiff

hisup/224/v3_image_vit_cnn_bs4x12/.hydra/hydra.yaml

@@ -112,15 +112,14 @@ hydra:
     hydra:
     - hydra.mode=RUN
     task:
-    - run_type=debug
-    - host=gin
     - checkpoint=best_val_iou
     - experiment=hisup_image
-    - +image_file=demo_data/image0_CH_val.tif
+    - +image_file=/user/rsulzer/home/inria4.tiff
+    - host=gin
   job:
     name: predict_demo
     chdir: null
-    override_dirname: +image_file=demo_data/image0_CH_val.tif,checkpoint=best_val_iou,experiment=hisup_image,host=gin,run_type=debug
+    override_dirname: +image_file=/user/rsulzer/home/inria4.tiff,checkpoint=best_val_iou,experiment=hisup_image,host=gin
     id: ???
     num: ???
     config_name: config
@@ -153,7 +152,7 @@ hydra:
       model@experiment.model: hisup
       encoder@experiment.encoder: vit_cnn
       dataset: p3
-      run_type: debug
+      run_type: release
       host: gin
       hydra/env: default
       hydra/callbacks: null

hisup/224/v3_image_vit_cnn_bs4x12/.hydra/overrides.yaml

@@ -1,5 +1,4 @@
-- run_type=debug
-- host=gin
 - checkpoint=best_val_iou
 - experiment=hisup_image
-- +image_file=demo_data/image0_CH_val.tif
+- +image_file=/user/rsulzer/home/inria4.tiff
+- host=gin

pix2poly/224/v4_image_vit_bs4x16/.hydra/config.yaml

@@ -7,14 +7,14 @@ host:
   update_pbar_every: 1
   ldof_exe: /user/rsulzer/home/cpp/line-DOF-metric/build/calculate_DoF
 run_type:
-  name: debug
+  name: release
   batch_size: 16
-  train_subset: 256
-  val_subset: 32
-  test_subset: 32
-  logging: DEBUG
-  num_workers: 0
-  log_to_wandb: false
+  train_subset: null
+  val_subset: null
+  test_subset: null
+  logging: INFO
+  num_workers: 16
+  log_to_wandb: true
 dataset:
   name: PixelsPointsPolygons
   size: ${..experiment.encoder.in_size}
@@ -103,8 +103,7 @@ evaluation:
   modes:
   - iou
   eval_file: results/metrics
-experiment.name: debug
 output_dir: ${.host.model_root}/${.experiment.model.name}/${.experiment.encoder.in_size}/${.experiment.name}
 checkpoint: best_val_iou
 num_workers: ${.run_type.num_workers}
-image_file: demo_data/image0_CH_val.tif
+image_file: /user/rsulzer/home/inria.tiff

pix2poly/224/v4_image_vit_bs4x16/.hydra/hydra.yaml

@@ -112,15 +112,14 @@ hydra:
     hydra:
     - hydra.mode=RUN
     task:
-    - run_type=debug
-    - host=gin
     - checkpoint=best_val_iou
     - experiment=p2p_image
-    - +image_file=demo_data/image0_CH_val.tif
+    - +image_file=/user/rsulzer/home/inria.tiff
+    - host=gin
   job:
     name: predict_demo
     chdir: null
-    override_dirname: +image_file=demo_data/image0_CH_val.tif,checkpoint=best_val_iou,experiment=p2p_image,host=gin,run_type=debug
+    override_dirname: +image_file=/user/rsulzer/home/inria.tiff,checkpoint=best_val_iou,experiment=p2p_image,host=gin
     id: ???
     num: ???
     config_name: config
@@ -153,7 +152,7 @@ hydra:
       model@experiment.model: pix2poly
       encoder@experiment.encoder: vit
       dataset: p3
-      run_type: debug
+      run_type: release
       host: gin
       hydra/env: default
       hydra/callbacks: null

pix2poly/224/v4_image_vit_bs4x16/.hydra/overrides.yaml

@@ -1,5 +1,4 @@
-- run_type=debug
-- host=gin
 - checkpoint=best_val_iou
 - experiment=p2p_image
-- +image_file=demo_data/image0_CH_val.tif
+- +image_file=/user/rsulzer/home/inria.tiff
+- host=gin