| 36 | writer.close() |
| 37 | |
| 38 | def get_frame_list(start_frame, end_frame): |
| 39 | # eg: |
| 40 | # start_frame: "000000000.jpg" |
| 41 | # end_frame: "000000039.jpg" |
| 42 | # return: ["000000000.jpg", "000000001.jpg", ..., "000000039.jpg"] |
| 43 | ext = start_frame.split('.')[-1] |
| 44 | len_num = len(start_frame.split('.')[0]) |
| 45 | start_frame = int(start_frame.split('.')[0]) |
| 46 | end_frame = int(end_frame.split('.')[0]) |
| 47 | frame_list = [] |
| 48 | for i in range(start_frame, end_frame + 1): |
| 49 | frame_list.append(str(i).zfill(len_num) + f'.{ext}') |
| 50 | return frame_list |
| 51 | |
| 52 | |
| 53 | def resize_for_rectangle_crop(arr, image_size, reshape_mode="random"): |