| """ |
| Find windows from a video with clip_ids. |
| |
| A window is defined by a [start_clip_idx, end_clip_idx] pair: |
| For example, assuming clip_len = 2 seconds |
| [0, 0] meaning a single clip window [0, 2] (seconds) |
| [10, 19] meaning a 9 clip window [20, 40] (seconds) |
| |
| """ |
|
|
|
|
| def convert_clip_ids_to_windows(clip_ids): |
| """ Inverse function of convert_windows_to_clip_ids |
| Args: |
| clip_ids: list(int), each is a index of a clip, starting from 0 |
| |
| Returns: |
| list(list(int)), each sublist contains two integers which are clip indices. |
| [10, 19] meaning a 9 clip window [20, 40] (seconds), if each clip is 2 seconds. |
| |
| >>> test_clip_ids = [56, 57, 58, 59, 60, 61, 62] + [64, ] + [67, 68, 69, 70, 71] |
| >>> convert_clip_ids_to_windows(test_clip_ids) |
| [[56, 62], [64, 64], [67, 71]] |
| """ |
| windows = [] |
| _window = [clip_ids[0], None] |
| last_clip_id = clip_ids[0] |
| for clip_id in clip_ids: |
| if clip_id - last_clip_id > 1: |
| _window[1] = last_clip_id |
| windows.append(_window) |
| _window = [clip_id, None] |
| last_clip_id = clip_id |
| _window[1] = last_clip_id |
| windows.append(_window) |
| return windows |
|
|
|
|
| def convert_windows_to_clip_ids(windows): |
| """ Inverse function of convert_clip_ids_to_windows |
| Args: |
| windows: list(list(int)), each sublist contains two integers which are clip indices. |
| [10, 11] meaning a 9 clip window [20, 40] (seconds), if each clip is 2 seconds. |
| |
| Returns: |
| clip_ids: list(int) |
| |
| >>> test_windows =[[56, 62], [64, 64], [67, 71]] |
| >>> convert_windows_to_clip_ids(test_windows) |
| [56, 57, 58, 59, 60, 61, 62] + [64, ] + [67, 68, 69, 70, 71] |
| """ |
| clip_ids = [] |
| for w in windows: |
| clip_ids += list(range(w[0], w[1]+1)) |
| return clip_ids |
|
|
|
|
| def convert_clip_window_to_seconds(window, clip_len=2): |
| return [window[0] * clip_len, (window[1] + 1) * clip_len] |
|
|
