Entry for running video anchor encoding.
(self, compress_dir)
| 577 | return trainset_list, valset_list |
| 578 | |
| 579 | def video_encode(self, compress_dir): |
| 580 | """Entry for running video anchor encoding.""" |
| 581 | print("Running video anchor encoding...") |
| 582 | |
| 583 | if os.path.exists(compress_dir): |
| 584 | shutil.rmtree(compress_dir) |
| 585 | os.makedirs(compress_dir) |
| 586 | |
| 587 | # transform splats_list |
| 588 | splats_list = self.compression_method.param_transform(self.splats_list, cfg.compression_cfg.transform_attributes) |
| 589 | # compression: loop on GOPs |
| 590 | num_gop = (self.frame_num + self.cfg.gop_size - 1) // self.cfg.gop_size |
| 591 | for gop_id in range(num_gop): |
| 592 | gop_start_frame_id = gop_id * self.cfg.gop_size |
| 593 | gop_end_frame_id = min(gop_start_frame_id + self.cfg.gop_size, self.frame_num) |
| 594 | gop_splats_list = splats_list[gop_start_frame_id:gop_end_frame_id] |
| 595 | self.compression_method.compress(gop_splats_list, compress_dir, gop_id) |
| 596 | |
| 597 | def video_decode(self, compress_dir): |
| 598 | """Entry for running video anchor decoding.""" |
no test coverage detected