Function main

examples/codec_ply_sequence.py:1108–1147

(local_rank: int, world_rank, world_size: int, cfg: Config)

    os.makedirs(path)

def main(local_rank: int, world_rank, world_size: int, cfg: Config):
    runner = Runner(local_rank, world_rank, world_size, cfg)
    
    render_stats = runner.eval('val')

    ### function test for transform and inverse transform
    # transform_stats = runner.test_transform()
    # runner.compare_render_stats(render_stats, transform_stats)
    compress_dir = f"{cfg.result_dir}/compression"
    if not cfg.decode_only:
        create_dir(compress_dir)

    if cfg.anchor_type == "video": # TODO: change "video" to "video_deprecated"
        splats_list_c = runner.compress(compress_dir)
    elif cfg.anchor_type == "video_codec": # TODO: change "video_codec" to "video"
        if not cfg.decode_only:
            runner.video_encode(compress_dir)
        splats_list_c = runner.video_decode(compress_dir)
    elif cfg.anchor_type == "pcc":
        splats_list_c = runner.pcc_compress(compress_dir)
    else:
        raise NotImplementedError(f"{cfg.anchor_type} Anchor has not been implemented.")
    
    # save the sequences of decoded gaussian splats into the sequences of ply files
    decoded_ply_dir = f"{cfg.result_dir}/decoded_ply"
    create_dir(decoded_ply_dir)
    for idx, splats_c in enumerate(splats_list_c):
        decoded_ply_filename = os.path.basename(runner.ply_filename_list[idx])
        save_ply(splats_c, os.path.join(decoded_ply_dir, decoded_ply_filename))
    
    for splats, splats_c in zip(runner.splats_list, splats_list_c):
        for k in splats.keys():
            splats[k].data = splats_c[k].to(runner.device)
           
    compress_stats = runner.eval(stage="compress")
    runner.compare_render_stats(render_stats, compress_stats, name1="Uncompressed", name2="Compressed")

    runner.stack_render_img_to_vid()
    runner.eval_pngs_with_gsc_ctc_metrics()
    # runner.summary()

if __name__ == "__main__":
    configs = {