File size: 679 Bytes
7df6a88 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | import os
os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'
import numpy as np
from datasets import load_dataset
import sklearn_submission
from triangulation import predict_wireframe_tracks
print("Loading dataset...")
dataset = load_dataset('usm3d/hoho22k_2026_trainval', split='train', streaming=True, trust_remote_code=True)
samples = [s for idx, s in enumerate(dataset) if idx < 3]
for idx, sample in enumerate(samples):
print(f"Sample {idx}")
try:
tv, te = predict_wireframe_tracks(sample)
print(f"Tracks: {len(tv)} vertices, {len(te)} edges")
except Exception as e:
import traceback
traceback.print_exc()
print(f"CRASH: {e}")
|