youngLBW / HRN

[CVPR2023] A Hierarchical Representation Network for Accurate and Detailed Face Reconstruction from In-The-Wild Images.
https://younglbw.github.io/HRN-homepage/
Apache License 2.0
407 stars 37 forks source link

list(list(float64)<iv=None>)<iv=None> cannot be represented as a Numpy dtype #50

Open liyanallen opened 8 months ago

liyanallen commented 8 months ago

023-10-24 09:48:59.912303: I tensorflow/core/platform/profile_utils/cpu_utils.cc:114] CPU Frequency: 1697970000 Hz predict ./assets/examples/single_view_image 0%| | 0/1 [00:07<?, ?it/s] Traceback (most recent call last): File "/home/ty/0_server_code/li/01.3dface/demo.py", line 78, in run_hrn(args) File "/home/ty/0_server_code/li/01.3dface/demo.py", line 26, in run_hrn output = reconstructor.predict(img, visualize=True, save_name=save_name, out_dir=out_dir) File "/home/ty/0_server_code/li/01.3dface/models/hrn.py", line 243, in predict output = self.predict_base(img) File "/home/ty/0_server_code/li/01.3dface/models/hrn.py", line 153, in predict_base landmarks = self.prepare_data(img, self.lm_sess, five_points=landmarks) File "/home/ty/0_server_code/li/01.3dface/models/hrn.py", line 98, in prepare_data landmark = lm_sess.get_landmarks_from_image(input_img)[0] File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context return func(*args, *kwargs) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/face_alignment/api.py", line 120, in get_landmarks_from_image detected_faces = self.face_detector.detect_from_image(image.copy()) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/face_alignment/detection/sfd/sfd_detector.py", line 44, in detect_from_image bboxlist = detect(self.face_detector, image, device=self.device)[0] File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/face_alignment/detection/sfd/detect.py", line 19, in detect return batch_detect(net, img, device) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/face_alignment/detection/sfd/detect.py", line 45, in batch_detect bboxlists = get_predictions(List(olist), batch_size) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/dispatcher.py", line 501, in _compile_for_args raise e File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/dispatcher.py", line 434, in _compile_for_args return_val = self.compile(tuple(argtypes)) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/dispatcher.py", line 979, in compile cres = self._compiler.compile(args, return_type) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/dispatcher.py", line 141, in compile status, retval = self._compile_cached(args, return_type) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/dispatcher.py", line 155, in _compile_cached retval = self._compile_core(args, return_type) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/dispatcher.py", line 168, in _compile_core cres = compiler.compile_extra(self.targetdescr.typing_context, File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/compiler.py", line 686, in compile_extra return pipeline.compile_extra(func) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/compiler.py", line 428, in compile_extra return self._compile_bytecode() File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/compiler.py", line 492, in _compile_bytecode return self._compile_core() File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/compiler.py", line 471, in _compile_core raise e File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/compiler.py", line 462, in _compile_core pm.run(self.state) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/compiler_machinery.py", line 343, in run raise patched_exception File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/compiler_machinery.py", line 334, in run self._runPass(idx, pass_inst, state) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/compiler_lock.py", line 35, in _acquire_compile_lock return func(args, **kwargs) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/compiler_machinery.py", line 289, in _runPass mutated |= check(pss.run_pass, internal_state) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/compiler_machinery.py", line 262, in check mangled = func(compiler_state) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/typed_passes.py", line 398, in run_pass lower.create_cpython_wrapper(flags.release_gil) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/lowering.py", line 247, in create_cpython_wrapper self.context.create_cpython_wrapper(self.library, self.fndesc, File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/cpu.py", line 183, in create_cpython_wrapper builder.build() File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/callwrapper.py", line 123, in build self.build_wrapper(api, builder, closure, args, kws) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/callwrapper.py", line 188, in build_wrapper obj = api.from_native_return(retty, retval, env_manager) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/pythonapi.py", line 1402, in from_native_return out = self.from_native_value(typ, val, env_manager) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/pythonapi.py", line 1416, in from_native_value return impl(typ, val, c) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/core/boxing.py", line 399, in box_array np_dtype = numpy_support.as_dtype(typ.dtype) File "/home/ty/1_server_software/anaconda3/envs/3dface/lib/python3.9/site-packages/numba/np/numpy_support.py", line 156, in as_dtype raise NotImplementedError("%r cannot be represented as a Numpy dtype" NotImplementedError: Failed in nopython mode pipeline (step: native lowering) list(list(float64)) cannot be represented as a Numpy dtype

liyanallen commented 8 months ago

Maybe numba's version is not right, i'm not sure

youngLBW commented 7 months ago

Hi, you can try to change the versions of numba or numpy. FYI, I am using numba==0.53.1, numpy==1.18.5.