How to get an updated albedo texture after compute_uvatlas() for TriangleMesh?

[aplatyps]

Checklist

• I have searched for similar issues.
• For Python issues, I have tested with the latest development wheel.
• I have checked the release documentation and the latest documentation (for main branch).

My Question

Hi, I'm trying to create a new UV atlas for my mesh. My mesh is already textured and comes with a texture map but I want to rearrange the uv with Open3D so that it looks tidier.

I wrote the following code:

mesh_ = o3d.io.read_triangle_mesh("path/to/mesh", enable_post_processing=True)
is_edge_manifold = mesh_.is_edge_manifold()
if not is_edge_manifold:
    tri_mesh = mesh_.remove_non_manifold_edges()
mesh = o3d.t.geometry.TriangleMesh.from_legacy(mesh_)
mesh.compute_uvatlas(size=1024, gutter=10, parallel_partitions=1)
o3d.t.io.write_triangle_mesh("path/to/output/mesh", mesh)

The original texture:

After running above code to get UV atlas, the texture become like this:

How do I align/update the texture uv with the existing albedo texture map?
mesh.zip