Function save_text

sat/sample_video.py:90–93

(text: str, save_path: str)

                writer.append_data(frame)

def save_text(text: str, save_path: str):
    txt_save_path = os.path.join(save_path, "text.txt")
    with open(txt_save_path, "w") as fout:
        fout.write(text)

def save_traj(traj: torch.tensor, save_path: str):
    # Write the traj in the end of the txt file