Pix2NeRF
Pix2NeRF copied to clipboard
Hi, I have re-train the pix2nerf with a single 3090 GPU on synchairs, but I have not achieve the view synthesis under the condition of a single view. If I...
RuntimeError: CUDA out of memory. Tried to allocate 768.00 MiB (GPU 0; 6.00 GiB total capacity; 3.51 GiB already allocated; 391.91 MiB free; 3.57 GiB reserved in total by PyTorch)
Hi there, I have used 2 T4 GPU to retrain the model using the train_con.py script. The issue(cleanup function) persists across 2 different datasets, CARLA and CelebA. I would like...
1. Render novel views of the given image: the source image:  the new view:  the reconstruction:  the hybrid optimization:   Does anyone know how to improve...
input an image and I want to output my own 3D model
 Hi, I have trained the celeba in a single 3090. But when I finish 30k iterations, the program reports KeyError: 'batch_size', do I need to modify the parameters? Would...
First of all, thank you for your work! When I used code to train my data, such problems occurred, which could not be solved. I hope you can make valuable...
(Pix2NeRF) bash-4.4$ CUDA_VISIBLE_DEVICES=0,1 python3 train_con.py --curriculum=celeba --output_dir='/data/yshan/data/Img/pix2NerfOutput' --dataset_dir='/data/yshan/data/Img/img_align_celeba' --encoder_type='CCS' --recon_lambda=5 --ssim_lambda=1 --vgg_lambda=1 --pos_lambda_gen=15 --lambda_e_latent=1 --lambda_e_pos=1 --cond_lambda=1 --load_encoder=1 Namespace(n_epochs=3000, sample_interval=1000, output_dir='/data/yshan/data/Img/pix2NerfOutput', load_dir='/data/yshan/data/Img/pix2NerfOutput', curriculum='celeba', eval_freq=5000, port='12354', set_step=None, model_save_interval=200, pretrained_dir='', wandb_name='', wandb_entity='', wandb_project='',...