InstantMesh icon indicating copy to clipboard operation
InstantMesh copied to clipboard

Train on nerf of stage 1 with custom dataset

Open hopeliu20160622 opened this issue 1 year ago • 10 comments

Hi, guys, i rendered my own rings dataset with 32 views , for every view, i have a RGBA image, where A means mask, normal image and depth image, and also saved the camera pos -- a 4*4 RT matrix. The size of the dataset is 8000. And i use the command to train: python train.py --base configs/instant-nerf-large-train.yaml --gpus 0,1,2,3,4,5,6,7 --num_nodes 1 After step = 7000, i checked the train images, as below, the third row, the two object has the same shape. Even i input a yellow duck multi-view images, the model output a ring like the third row. train_0001000 Can anyone help? Thanks. That's really weird.

hopeliu20160622 avatar Jun 19 '24 08:06 hopeliu20160622

I'm sorry I have no idea on this problem.

bluestyle97 avatar Jun 21 '24 14:06 bluestyle97

@hopeliu20160622 hello,how did you render your datasets with 32 views?Use python code?i am interested in it .!!

throb081 avatar Jun 30 '24 03:06 throb081

I'm sorry I have no idea on this problem.

Thanks , can you share a training sample, I think there is something wrong with my render script, maybe i can compare the details to find the difference. Thanks. [email protected]

Hello,I am interested in render script for preparing training dateset, cloud you share me it? Thanks, [email protected]

ustbzgn avatar Jul 02 '24 06:07 ustbzgn

I'm sorry I have no idea on this problem.

Thanks , can you share a training sample, I think there is something wrong with my render script, maybe i can compare the details to find the difference. Thanks. [email protected]

Hello,I am interested in render script for preparing training dateset, cloud you share me it? Thanks, [email protected]

guangxuwang avatar Jul 05 '24 09:07 guangxuwang

@hopeliu20160622 Hi, I already made the dataset, but I don't know how to add my own dataset in 'instant-nerf-large-train.yaml', can you share your config file? My dataset is of the following type for your convenience.Where in pose is the npz file of 32 viewpoints. In rgba there are 32 views (png type),Looking forward to your reply ,[email protected] 微信图片_20240714230249

Mrguanglei avatar Jul 14 '24 15:07 Mrguanglei

Hello, I'm also interested in the rendering scripts for rendering the training dataset. [email protected], Looking forward to your reply.

zkaiWu avatar Jul 17 '24 07:07 zkaiWu

Can anybody post the rendering scripts here? It helps a lot of people. Thanks!

denred0 avatar Jul 17 '24 08:07 denred0

Hi, guys, i rendered my own rings dataset with 32 views , for every view, i have a RGBA image, where A means mask, normal image and depth image, and also saved the camera pos -- a 4*4 RT matrix. The size of the dataset is 8000. And i use the command to train: python train.py --base configs/instant-nerf-large-train.yaml --gpus 0,1,2,3,4,5,6,7 --num_nodes 1 After step = 7000, i checked the train images, as below, the third row, the two object has the same shape. Even i input a yellow duck multi-view images, the model output a ring like the third row. train_0001000 Can anyone help? Thanks. That's really weird.

Hi @hopeliu20160622, may I know did you load in the openlrm checkpoint for training instantnerf? It seems like there are several tensor shape unmatching cases here.

HaFred avatar Aug 09 '24 00:08 HaFred

Hi, guys, i rendered my own rings dataset with 32 views , for every view, i have a RGBA image, where A means mask, normal image and depth image, and also saved the camera pos -- a 4*4 RT matrix. The size of the dataset is 8000. And i use the command to train: python train.py --base configs/instant-nerf-large-train.yaml --gpus 0,1,2,3,4,5,6,7 --num_nodes 1 After step = 7000, i checked the train images, as below, the third row, the two object has the same shape. Even i input a yellow duck multi-view images, the model output a ring like the third row. train_0001000 Can anyone help? Thanks. That's really weird.

Did you figure out this problem? Now I am fine-tuning the NeRF model, but cannot run due to 'cuda out of memory'. I set batch size =1, and use 2x L40 VRAM 48G

Jinyiyi3 avatar Mar 27 '25 09:03 Jinyiyi3