Views
No views yet
pip install hoho2025pip install git+http://hf.co/usm3d/tools2025.git 1git clone http://hf.co/usm3d/tools2025
2cd tools2025
3pip install -e .1from datasets import load_dataset
2from hoho2025.vis import plot_all_modalities
3from hoho2025.viz3d import *
4
5def read_colmap_rec(colmap_data):
6 import pycolmap
7 import tempfile,zipfile
8 import io
9 with tempfile.TemporaryDirectory() as tmpdir:
10 with zipfile.ZipFile(io.BytesIO(colmap_data), "r") as zf:
11 zf.extractall(tmpdir) # unpacks cameras.txt, images.txt, etc. to tmpdir
12 # Now parse with pycolmap
13 rec = pycolmap.Reconstruction(tmpdir)
14 return rec
15
16ds = load_dataset("usm3d/hoho22k_2026_trainval", streaming=True, trust_remote_code=True)
17for a in ds['train']:
18 break
19
20fig, ax = plot_all_modalities(a)
21
22## Now 3d
23
24fig3d = init_figure()
25plot_reconstruction(fig3d, read_colmap_rec(a['colmap']))
26plot_wireframe(fig3d, a['wf_vertices'], a['wf_edges'], a['wf_classifications'])
27plot_bpo_cameras_from_entry(fig3d, a)
28fig3d1from hoho2025.example_solutions import predict_wireframe
2pred_vertices, pred_connections = predict_wireframe(a)
3
4fig3d = init_figure()
5plot_reconstruction(fig3d, read_colmap_rec(a['colmap']))
6plot_wireframe(fig3d, pred_vertices, pred_connections, color='rgb(0, 0, 255)')
7fig3d1from hoho2025.metric_helper import hss
2
3score = hss(pred_vertices, pred_connections, a['wf_vertices'], a['wf_edges'], vert_thresh=0.5, edge_thresh=0.5)
4print (score)