czyoung commited on
Commit
24dc2fc
·
verified ·
1 Parent(s): 4249b90

Update calcSpeakingTypes to utilize new annotationToNoiseList

Browse files
Files changed (1) hide show
  1. sonogram_utility.py +6 -3
sonogram_utility.py CHANGED
@@ -413,11 +413,14 @@ def calcSpeakingTypes(pipeline,myAnnotation,maxTime):
413
  categorySegmentList, timeSteps = pipeline.annotationToNoiseList(myAnnotation)
414
  # [group,individual,silence], each as (start,duration)
415
  for seg in categorySegmentList[0]:
416
- mvAnnotation[Segment(seg[0],seg[0]+seg[1])] = 'group'
 
 
 
417
  for seg in categorySegmentList[1]:
418
- ovAnnotation[Segment(seg[0],seg[0]+seg[1])] = 'individual'
419
  for seg in categorySegmentList[2]:
420
- nvAnnotation[Segment(seg[0],seg[0]+seg[1])] = 'silence'
421
  return nvAnnotation, ovAnnotation, mvAnnotation
422
 
423
  def timeToString(timeInSeconds):
 
413
  categorySegmentList, timeSteps = pipeline.annotationToNoiseList(myAnnotation)
414
  # [group,individual,silence], each as (start,duration)
415
  for seg in categorySegmentList[0]:
416
+ if seg[0] == 'group':
417
+ mvAnnotation[Segment(seg[0],seg[0]+seg[1])] = 'unclear'
418
+ else:
419
+ mvAnnotation[seg[1]] = seg[0]
420
  for seg in categorySegmentList[1]:
421
+ ovAnnotation[seg[1]] = seg[0]
422
  for seg in categorySegmentList[2]:
423
+ nvAnnotation[seg[1]] = 'silence'
424
  return nvAnnotation, ovAnnotation, mvAnnotation
425
 
426
  def timeToString(timeInSeconds):