INNER CODE UNIT · Python
first
htdt/godogen · asset-gen/tools/find_loop_frame.py:63
first = skip + window
last = n - window
if first > last:
return None, 0.0, []
# Score all candidates
all_candidates = []
for start in range(first, last + 1):
sim = window_similarity(embeddings, 0, start, window)
all_candidates.append((start, sim))
# Take top K by similarity
top = sorted(all_candidates, key=lambda c: -c[1])[:TOP_K]
# Deduplicate nearby frames
peaks = dedupe(top, min_gap)
if not peaks: