search

ayaanzhaque / instruct-nerf2nerf

Instruct-NeRF2NeRF: Editing 3D Scenes with Instructions (ICCV 2023)
https://instruct-nerf2nerf.github.io/
MIT License
769 stars 64 forks source link

How to save the modified image dataset? #67

Closed w2690831985w closed 8 months ago

w2690831985w commented 9 months ago

After running in2n, I got a modified nerf model ./outputs/xxx/in2n/xxx/nerfstudio_models/step-000055000.ckpt. It did modified the nerf model successfully. However, there is no modified image dataset in this folder, and the in2n command did not modify the original image dataset used to train nerfacto in the ./data folder. Also, I did not see any modification in the image dataset displyed in the nerfstudio viewer.

So, where can the modified image dataset be displayed? What if I want to save it?

ttaa9 commented 8 months ago

@ayaanzhaque Is this already implemented somewhere?

ayaanzhaque commented 8 months ago

Hi, so the modified dataset is not automatically printed. If you would like to print it, you can access the tensor of images in in2n_pipeline.py with the following: self.datamanager.image_batch["image"]

Then you can just print the images however you would like.