Joshua Lochner commited on
Commit
aa018be
·
1 Parent(s): 87b2dec

Do not allow predictions to miss start of video

Browse files
Files changed (1) hide show
  1. src/predict.py +4 -1
src/predict.py CHANGED
@@ -166,6 +166,9 @@ SEGMENT_MATCH_RE = fr'{_SEGMENT_START}\s*(?P<text>.*?)\s*(?:{_SEGMENT_END}|$)'
166
  MATCH_WINDOW = 25 # Increase for accuracy, but takes longer: O(n^3)
167
  MERGE_TIME_WITHIN = 8 # Merge predictions if they are within x seconds
168
 
 
 
 
169
 
170
  @dataclass(frozen=True, eq=True)
171
  class ClassifierArguments:
@@ -367,7 +370,7 @@ def segments_to_predictions(segments, model, tokenizer):
367
 
368
  final_predicted_time_ranges = []
369
  for range in predicted_time_ranges:
370
- start_time = range['start']
371
  end_time = range['end']
372
 
373
  if prev_prediction is not None and \
 
166
  MATCH_WINDOW = 25 # Increase for accuracy, but takes longer: O(n^3)
167
  MERGE_TIME_WITHIN = 8 # Merge predictions if they are within x seconds
168
 
169
+ # Any prediction whose start time is <= this will be set to start at 0
170
+ START_TIME_ZERO_THRESHOLD = 0.08
171
+
172
 
173
  @dataclass(frozen=True, eq=True)
174
  class ClassifierArguments:
 
370
 
371
  final_predicted_time_ranges = []
372
  for range in predicted_time_ranges:
373
+ start_time = range['start'] if range['start'] > START_TIME_ZERO_THRESHOLD else 0
374
  end_time = range['end']
375
 
376
  if prev_prediction is not None and \