| 116 | |
| 117 | |
| 118 | class FrameSampler: |
| 119 | def __init__(self, max_num_frames=-1, min_num_frames=8, sample='rand'): |
| 120 | self.max_num_frames = max_num_frames |
| 121 | self.min_num_frames = min_num_frames |
| 122 | self.sample = sample |
| 123 | |
| 124 | def __call__(self, file_name): |
| 125 | fn = read_frames_folder if file_name.endswith('/') else read_frames_decord |
| 126 | frames = fn(file_name, num_frames=self.max_num_frames, min_num_frames=self.min_num_frames, sample=self.sample) |
| 127 | return frames |
| 128 | |
| 129 | |
| 130 | def decode_video_byte(video_bytes): |
no outgoing calls
no test coverage detected