scenediffuser / Scene-Diffuser

Official implementation of CVPR23 paper "Diffusion-based Generation, Optimization, and Planning in 3D Scenes"
https://scenediffuser.github.io/
MIT License
351 stars 24 forks source link

Question about the motion generation. #20

Open xjchao7 opened 6 days ago

xjchao7 commented 6 days ago

I wanna ask could i change the start pose in the motion generation test process? How could i change the start pose? Thank you very much.

Silverster98 commented 3 days ago

You should modify the dataloader. Our code directly loads motion from PROX and segments them into fixed horizons, then uses the first frame as the condition. The key code is here: https://github.com/scenediffuser/Scene-Diffuser/blob/main/datasets/transforms.py#L64-L66

If you want to change the start pose, the simplest way may be to directly change this data[start] using your pose.