(self, real_len, clip_len=81, overlap=1)
| 213 | return target_array[:target_len] |
| 214 | |
| 215 | def get_valid_len(self, real_len, clip_len=81, overlap=1): |
| 216 | real_clip_len = clip_len - overlap |
| 217 | last_clip_num = (real_len - overlap) % real_clip_len |
| 218 | if last_clip_num == 0: |
| 219 | extra = 0 |
| 220 | else: |
| 221 | extra = real_clip_len - last_clip_num |
| 222 | target_len = real_len + extra |
| 223 | return target_len |
| 224 | |
| 225 | |
| 226 | def get_i2v_mask(self, lat_t, lat_h, lat_w, mask_len=1, mask_pixel_values=None, device="cuda"): |