ReassembleNet: Learnable Keypoints and Diffusion for 2D Fresco Reconstruction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Islam, Adeela, Fiorini, Stefano, James, Stuart, Morerio, Pietro, Del Bue, Alessio
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917091168223232
author Islam, Adeela
Fiorini, Stefano
James, Stuart
Morerio, Pietro
Del Bue, Alessio
author_facet Islam, Adeela
Fiorini, Stefano
James, Stuart
Morerio, Pietro
Del Bue, Alessio
contents The task of reassembly is a significant challenge across multiple domains, including archaeology, genomics, and molecular docking, requiring the precise placement and orientation of elements to reconstruct an original structure. In this work, we address key limitations in state-of-the-art Deep Learning methods for reassembly, namely i) scalability; ii) multimodality; and iii) real-world applicability: beyond square or simple geometric shapes, realistic and complex erosion, or other real-world problems. We propose ReassembleNet, a method that reduces complexity by representing each input piece as a set of contour keypoints and learning to select the most informative ones by Graph Neural Networks pooling inspired techniques. ReassembleNet effectively lowers computational complexity while enabling the integration of features from multiple modalities, including both geometric and texture data. Further enhanced through pretraining on a semi-synthetic dataset. We then apply diffusion-based pose estimation to recover the original structure. We improve on prior methods by 57% and 87% for RMSE Rotation and Translation, respectively.
format Preprint
id arxiv_https___arxiv_org_abs_2505_21117
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle ReassembleNet: Learnable Keypoints and Diffusion for 2D Fresco Reconstruction
Islam, Adeela
Fiorini, Stefano
James, Stuart
Morerio, Pietro
Del Bue, Alessio
Computer Vision and Pattern Recognition
The task of reassembly is a significant challenge across multiple domains, including archaeology, genomics, and molecular docking, requiring the precise placement and orientation of elements to reconstruct an original structure. In this work, we address key limitations in state-of-the-art Deep Learning methods for reassembly, namely i) scalability; ii) multimodality; and iii) real-world applicability: beyond square or simple geometric shapes, realistic and complex erosion, or other real-world problems. We propose ReassembleNet, a method that reduces complexity by representing each input piece as a set of contour keypoints and learning to select the most informative ones by Graph Neural Networks pooling inspired techniques. ReassembleNet effectively lowers computational complexity while enabling the integration of features from multiple modalities, including both geometric and texture data. Further enhanced through pretraining on a semi-synthetic dataset. We then apply diffusion-based pose estimation to recover the original structure. We improve on prior methods by 57% and 87% for RMSE Rotation and Translation, respectively.
title ReassembleNet: Learnable Keypoints and Diffusion for 2D Fresco Reconstruction
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2505.21117