(frame_idx)
| 576 | return dir_path |
| 577 | |
| 578 | def _save_frame(frame_idx): |
| 579 | frame_dict = {} |
| 580 | for key in seq_keys: |
| 581 | frame_dict[key] = predictions[key][frame_idx] |
| 582 | np.savez(os.path.join(dir_path, f"frame_{frame_idx:06d}.npz"), **frame_dict) |
| 583 | |
| 584 | num_workers = min(32, S) |
| 585 | with ThreadPoolExecutor(max_workers=num_workers) as pool: |
nothing calls this directly
no outgoing calls
no test coverage detected