pyannote_core_segments_annotations_temporal_operations_quickstart.py

1from pyannote.core import Segment, Annotation, Timeline
2
3# 1. Create a temporal segment [start, end]
4segment = Segment(start=2.3, end=4.8)
5print(f"Segment duration: {segment.duration:.1f}s")
6
7# 2. Create an annotation (a collection of labeled segments)
8annotation = Annotation()
9annotation[Segment(1, 5)] = 'Alice'
10annotation[Segment(3, 8)] = 'Bob'
11annotation[Segment(12, 13)] = 'Alice'
12
13# 3. Perform basic operations
14# Get the timeline (set of segments without labels)
15timeline = annotation.get_timeline()
16
17# Find where 'Alice' and 'Bob' are speaking at the same time
18overlap = annotation.label_overlap('Alice', 'Bob')
19
20# 4. Print results
21print(f"Total duration: {annotation.get_timeline().duration():.1f}s")
22print(f"Labels: {annotation.labels()}")
23
24for segment, track, label in annotation.itertracks(yield_label=True):
25    print(f"[{segment.start:4.1f}s --> {segment.end:4.1f}s] {label}")