yapengyu
commented
3 years ago Then I modify neural_renderer/look.py similar to look_at.py, liking:
def look(vertices, eye, direction=[0, 1, 0], up=None):
def look(vertices, eye, direction=[0, 1, 0], up=[0, 1, 0]): .... if isinstance(up, list) or isinstance(up, tuple): up = torch.tensor(up, dtype=torch.float32, device=device) elif isinstance(up, np.ndarray): up = torch.from_numpy(up).to(device) elif torch.is_tensor(up): up.to(device)
new Error appeared:
dataset [AllFaceDataset] of size 8 was created Testing gpu [0] Network [RotateSPADEGenerator] was created. Total number of parameters: 225.1 million. To see the architecture, do print(network). start prefetching data... Process Process-1: Traceback (most recent call last): File "/ssd/anaconda3/envs/rr/lib/python3.6/multiprocessing/process.py", line 258, in _bootstrap self.run() File "/ssd/anaconda3/envs/rr/lib/python3.6/multiprocessing/process.py", line 93, in run self._target(*self._args, *self._kwargs) File "/ssd/experiment/Rotate-and-Render/data/data_utils.py", line 146, in prefetch_data prefetcher = data_prefetcher(dataloader, opt, render_layer) File "/ssd/experiment/Rotate-and-Render/data/data_utils.py", line 99, in init self.preload() File "/ssd/experiment/Rotate-and-Render/data/data_utils.py", line 124, in preload self.next_input = get_multipose_test_input(data, self.render_layer, self.opt.yaw_poses, self.opt.pitch_poses) File "/ssd/experiment/Rotate-and-Render/data/data_utils.py", line 65, in get_multipose_test_input = render.rotate_render(data['param_path'], real_image, data['M'], yaw_pose=pose) File "/ssd/experiment/Rotate-and-Render/models/networks/rotate_render.py", line 80, in rotate_render rendered_images, depths, masks, = self.renderer(vertices_ori_normal, self.faces_use, texs) # rendered_images: batch 3 h w, masks: batch h w ValueError: not enough values to unpack (expected 3, got 1)
MingzSu
JunYoungOH97
dataset [AllFaceDataset] of size 8 was created Testing gpu [0] Network [RotateSPADEGenerator] was created. Total number of parameters: 225.1 million. To see the architecture, do print(network). start prefetching data... Process Process-1: Traceback (most recent call last): File "/ssd/anaconda3/envs/rr/lib/python3.6/multiprocessing/process.py", line 258, in _bootstrap self.run() File "/ssd/anaconda3/envs/rr/lib/python3.6/multiprocessing/process.py", line 93, in run self._target(*self._args, self._kwargs) File "/ssd/experiment/Rotate-and-Render/data/data_utils.py", line 146, in prefetch_data prefetcher = data_prefetcher(dataloader, opt, render_layer) File "/ssd/experiment/Rotate-and-Render/data/data_utils.py", line 99, in init self.preload() File "/ssd/experiment/Rotate-and-Render/data/data_utils.py", line 124, in preload self.next_input = get_multipose_test_input(data, self.render_layer, self.opt.yaw_poses, self.opt.pitch_poses) File "/ssd/experiment/Rotate-and-Render/data/data_utils.py", line 65, in get_multipose_test_input = render.rotate_render(data['param_path'], real_image, data['M'], yaw_pose=pose) File "/ssd/experiment/Rotate-and-Render/models/networks/rotate_render.py", line 80, in rotate_render rendered_images, depths, masks, = self.renderer(vertices_ori_normal, self.faces_use, texs) # rendered_images: batch 3 h w, masks: batch h w File "/ssd/anaconda3/envs/rr/lib/python3.6/site-packages/torch/nn/modules/module.py", line 889, in _call_impl result = self.forward(input, kwargs) File "/ssd/anaconda3/envs/rr/lib/python3.6/site-packages/neural_renderer-1.1.3-py3.6-linux-x86_64.egg/neural_renderer/renderer.py", line 68, in forward return self.render(vertices, faces, textures) File "/ssd/anaconda3/envs/rr/lib/python3.6/site-packages/neural_renderer-1.1.3-py3.6-linux-x86_64.egg/neural_renderer/renderer.py", line 148, in render vertices = nr.look(vertices, self.eye, self.camera_direction) File "/ssd/anaconda3/envs/rr/lib/python3.6/site-packages/neural_renderer-1.1.3-py3.6-linux-x86_64.egg/neural_renderer/look.py", line 41, in look if up.ndimension() == 1: AttributeError: 'NoneType' object has no attribute 'ndimension'