kwea123 / nerf_pl

NeRF (Neural Radiance Fields) and NeRF in the Wild using pytorch-lightning
https://www.youtube.com/playlist?list=PLDV2CyUo4q-K02pNEyDr7DYpTQuka3mbV
MIT License
2.74k stars 483 forks source link

Input size while running "360 inward-facing scene" on custom data #156

Open imadgohar opened 2 years ago

imadgohar commented 2 years ago

Describe the bug What should be the size of image, my input image dimension is 5280*2970.

Which branch you use Training on Custom Data, Executing "360 inward-facing scene" in nerp.pl

When I use the default image size, It just stop working after 3 to 5 minutes. On 512*512 the error is shown Capture

/content/nerf_pl INFO:numexpr.utils:NumExpr defaulting to 2 threads. INFO:lightning:GPU available: True, used: True INFO:lightning:CUDA_VISIBLE_DEVICES: [0] Traceback (most recent call last): File "train.py", line 180, in trainer.fit(system) File "/usr/local/lib/python3.7/dist-packages/pytorch_lightning/trainer/trainer.py", line 730, in fit model.prepare_data() File "train.py", line 80, in prepare_data self.train_dataset = dataset(split='train', **kwargs) File "/content/nerf_pl/datasets/llff.py", line 173, in init self.read_meta() File "/content/nerf_pl/datasets/llff.py", line 191, in read_meta f'You must set @img_wh to have the same aspect ratio as ({W}, {H}) !' AssertionError: You must set @img_wh to have the same aspect ratio as (5280.0, 2970.0) !