Upload diarization_pyannote_gilbert.py with huggingface_hub
Browse files
diarization_pyannote_gilbert.py
CHANGED
|
@@ -287,13 +287,17 @@ def run_gilbert_diarization(
|
|
| 287 |
# Convertir en segments
|
| 288 |
segments = annotation_to_segments(annotation)
|
| 289 |
|
| 290 |
-
# Post-traitement Gilbert
|
| 291 |
-
|
| 292 |
-
|
| 293 |
-
|
| 294 |
-
|
| 295 |
-
|
| 296 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 297 |
|
| 298 |
# Détecter les overlaps
|
| 299 |
overlaps = detect_overlaps_gilbert(segments_processed)
|
|
@@ -401,8 +405,8 @@ def main():
|
|
| 401 |
parser.add_argument("--max_speakers", type=int, default=None, help="Nombre maximum de locuteurs")
|
| 402 |
parser.add_argument("--exclusive", action="store_true", help="Utiliser exclusive_speaker_diarization")
|
| 403 |
parser.add_argument("--no-progress", action="store_true", help="Ne pas afficher la progression")
|
| 404 |
-
parser.add_argument("--min-segment", type=float, default=0.
|
| 405 |
-
parser.add_argument("--merge-gaps", type=float, default=0.
|
| 406 |
|
| 407 |
args = parser.parse_args()
|
| 408 |
|
|
|
|
| 287 |
# Convertir en segments
|
| 288 |
segments = annotation_to_segments(annotation)
|
| 289 |
|
| 290 |
+
# Post-traitement Gilbert (optionnel - peut être désactivé)
|
| 291 |
+
if min_segment_duration > 0 or merge_gaps > 0:
|
| 292 |
+
print("[Gilbert] Post-traitement intelligent des segments...")
|
| 293 |
+
segments_processed = post_process_segments_gilbert(
|
| 294 |
+
segments,
|
| 295 |
+
min_segment_duration=min_segment_duration,
|
| 296 |
+
max_gap=merge_gaps
|
| 297 |
+
)
|
| 298 |
+
else:
|
| 299 |
+
print("[Gilbert] Post-traitement désactivé (segments bruts conservés)")
|
| 300 |
+
segments_processed = segments
|
| 301 |
|
| 302 |
# Détecter les overlaps
|
| 303 |
overlaps = detect_overlaps_gilbert(segments_processed)
|
|
|
|
| 405 |
parser.add_argument("--max_speakers", type=int, default=None, help="Nombre maximum de locuteurs")
|
| 406 |
parser.add_argument("--exclusive", action="store_true", help="Utiliser exclusive_speaker_diarization")
|
| 407 |
parser.add_argument("--no-progress", action="store_true", help="Ne pas afficher la progression")
|
| 408 |
+
parser.add_argument("--min-segment", type=float, default=0.0, help="Durée minimale des segments (s). 0 = désactivé (recommandé pour meilleure précision)")
|
| 409 |
+
parser.add_argument("--merge-gaps", type=float, default=0.0, help="Gaps à fusionner (s). 0 = désactivé (recommandé pour meilleure précision)")
|
| 410 |
|
| 411 |
args = parser.parse_args()
|
| 412 |
|