MEscriva commited on
Commit
c62be5a
·
verified ·
1 Parent(s): 9c3c159

Upload diarization_pyannote_gilbert.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. diarization_pyannote_gilbert.py +13 -9
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
- print("[Gilbert] Post-traitement intelligent des segments...")
292
- segments_processed = post_process_segments_gilbert(
293
- segments,
294
- min_segment_duration=min_segment_duration,
295
- max_gap=merge_gaps
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.5, help="Durée minimale des segments (s)")
405
- parser.add_argument("--merge-gaps", type=float, default=0.3, help="Gaps à fusionner (s)")
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