MrTornado24 / Next3D

[CVPR 2023 Highlight] Next3D: Generative Neural Texture Rasterization for 3D-Aware Head Avatars
https://mrtornado24.github.io/Next3D/
470 stars 29 forks source link

TypeError: expected size to be one of int or Tuple[int] or Tuple[int, int] or Tuple[int, int, int], but got size with types [<class 'numpy.int64'>, <class 'numpy.int64'>] #37

Open liuhw01 opened 6 months ago

liuhw01 commented 6 months ago

When I load the network from pretrained_models/next3d_ffhq_512.pkl,

Running img = G.synthesis() in all the code results in an error:

_ = G.synthesis(ws[:1], c[:1], verts[:1]) # warm up

File "", line 154, in synthesis File "/root/miniconda3/envs/next3d/lib/python3.9/site-packages/torch/nn/functional.py", line 3924, in interpolate raise TypeError( TypeError: expected size to be one of int or Tuple[int] or Tuple[int, int] or Tuple[int, int, int], but got size with types [<class 'numpy.int64'>, <class 'numpy.int64'>]

What could be the reason for this?

zjh21 commented 3 months ago

same problem here