Export mesh problem

[yufeng9819]

Hi!

Thanks for your wonderful work. Now, I am trying to export mesh from training result through the following command "python launch.py --config configs/consistent4d.yaml --export --gpu 1 data.image_seq_path=./load/demo_pac_nerf/letter system.exporter_type=mesh-exporter system.exporter.fmt=obj". However, I encounter the following error.

I check the code in ./Consistent4D-main/threestudio/models/geometry/cascade_kplanes.py and find that it did miss the required arguments "timestamps". So, I want to ask how to add the missing arguments to the function "forward_density".

Looking forward to your reply.

[yanqinJiang]

Currently only per-frame mesh could be exported, referring to #5 For your case, a simple solution is

1. Set timesteps in forward_densify as keyword argument so that we don't need to pass the parameters.
2. Add export_timestep in configs here and here, the value should between [-1, 1]

3. Modify the forward_densify function:

   def forward_density(self, points: Float[Tensor, "*N Di"], timestamps: Float[Tensor, "*N 1"]=None, occ_eval=False) -> Float[Tensor, "*N 1"]:
       points_unscaled = points
       points = contract_to_unisphere(points_unscaled, self.bbox, self.unbounded)
       points = points * 2 - 1  # convert to [-1, 1] for grid sample
   
      if timestamps is None:
         timestamps = torch.ones_like(points[..., 0:1]) * self.cfg.export_timestamp
   
       points_input = torch.cat([points, timestamps], dim=-1) # timestamps range [-1, 1]
   
       # the rest lines remain the same
       # 

Please have a try, and if you meet any problem, feel free to ask me for help. (I didn't tested this code by myself)