dunbar12138 / pix2pix3D

pix2pix3D: Generating 3D Objects from 2D User Inputs
http://www.cs.cmu.edu/~pix2pix3D/
MIT License
1.65k stars 132 forks source link

raise AssertionError(f'Wrong size for dimension {idx}: got {size}, expected {ref_size}') AssertionError: Wrong size for dimension 1: got 0, expected 25 #35

Open manoj-samal opened 6 months ago

manoj-samal commented 6 months ago

!python applications/generate_samples.py --network checkpoints/pix2pix3d_seg2cat.pkl --outdir examples --random_seed 1 --cfg seg2cat --input_id 2

was running in google colab

Traceback (most recent call last): File "/content/pix2pix3D/applications/generate_samples.py", line 128, in main() File "/content/pix2pix3D/applications/generate_samples.py", line 113, in main ws = G.mapping(z, input_pose, {'mask': input_label, 'pose': input_pose}) File "", line 919, in mapping File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl return self._call_impl(*args, *kwargs) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1520, in _call_impl return forward_call(args, **kwargs) File "", line 361, in forward File "/content/pix2pix3D/./torch_utils/misc.py", line 97, in assert_shape raise AssertionError(f'Wrong size for dimension {idx}: got {size}, expected {ref_size}') AssertionError: Wrong size for dimension 1: got 0, expected 25

raise AssertionError(f'Wrong size for dimension {idx}: got {size}, expected {ref_size}') AssertionError: Wrong size for dimension 1: got 0, expected 25