Joshua Lochner commited on
Commit
183ba5e
1 Parent(s): 11be1bd

Output auto-submission link for missing segments

Browse files
Files changed (1) hide show
  1. src/evaluate.py +14 -2
src/evaluate.py CHANGED
@@ -18,6 +18,7 @@ import json
18
  import os
19
  import random
20
  from shared import seconds_to_time
 
21
 
22
 
23
  @dataclass
@@ -225,11 +226,12 @@ def main():
225
  seg for seg in labelled_predicted_segments if seg['best_prediction'] is None]
226
 
227
  if missed_segments or incorrect_segments:
228
- print('Issues identified for',
229
- video_id, f'(#{video_index})')
230
  # Potentially missed segments (model predicted, but not in database)
231
  if missed_segments:
232
  print(' - Missed segments:')
 
233
  for i, missed_segment in enumerate(missed_segments, start=1):
234
  print(f'\t#{i}:', seconds_to_time(
235
  missed_segment['start']), '-->', seconds_to_time(missed_segment['end']))
@@ -240,6 +242,16 @@ def main():
240
  print('\t\tProbability:',
241
  missed_segment.get('probability'))
242
 
 
 
 
 
 
 
 
 
 
 
243
  # Potentially incorrect segments (model didn't predict, but in database)
244
  if incorrect_segments:
245
  print(' - Incorrect segments:')
 
18
  import os
19
  import random
20
  from shared import seconds_to_time
21
+ from urllib.parse import quote
22
 
23
 
24
  @dataclass
 
226
  seg for seg in labelled_predicted_segments if seg['best_prediction'] is None]
227
 
228
  if missed_segments or incorrect_segments:
229
+ print(
230
+ f'Issues identified for https://youtu.be/{video_id} (#{video_index})')
231
  # Potentially missed segments (model predicted, but not in database)
232
  if missed_segments:
233
  print(' - Missed segments:')
234
+ segments_to_submit = []
235
  for i, missed_segment in enumerate(missed_segments, start=1):
236
  print(f'\t#{i}:', seconds_to_time(
237
  missed_segment['start']), '-->', seconds_to_time(missed_segment['end']))
 
242
  print('\t\tProbability:',
243
  missed_segment.get('probability'))
244
 
245
+ segments_to_submit.append({
246
+ 'segment': [missed_segment['start'], missed_segment['end']],
247
+ 'category': missed_segment['category'].lower(),
248
+ 'actionType': 'skip'
249
+ })
250
+
251
+ json_data = quote(json.dumps(segments_to_submit))
252
+ print(
253
+ f'\tSubmit: https://www.youtube.com/watch?v={video_id}#segments={json_data}')
254
+
255
  # Potentially incorrect segments (model didn't predict, but in database)
256
  if incorrect_segments:
257
  print(' - Incorrect segments:')