search

thohemp / 6DRepNet

Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.
MIT License
550 stars 72 forks source link

Pip install failed #29

Closed GlennCGL closed 2 years ago

GlennCGL commented 2 years ago

When I install the package: pip install SixDRepNet, it returns Collecting SixDRepNet Downloading SixDRepNet-0.1.1.tar.gz (23 kB) Preparing metadata (setup.py) ... error error: subprocess-exited-with-error

× python setup.py egg_info did not run successfully. │ exit code: 1 ╰─> [6 lines of output] Traceback (most recent call last): File "", line 36, in File "", line 34, in File "/private/var/folders/5l/9fsdwp_n1td91scw10n67zwc0000gp/T/pip-install-8xwcmbp1/sixdrepnet_2b06c43f46c5428d9c99677633d23b6e/setup.py", line 23, in long_description="".join(open("README.MD", "r").readlines()), FileNotFoundError: [Errno 2] No such file or directory: 'README.MD' [end of output]

note: This error originates from a subprocess, and is likely not a problem with pip. error: metadata-generation-failed

× Encountered error while generating package metadata. ╰─> See above for output.

note: This is an issue with the package mentioned above, not pip. hint: See above for details.

GlennCGL commented 2 years ago

And demo can not run successfully: python SixDRepNet/demo.py --snapshot 6DRepNet_300W_LP_AFLW2000.pth --cam 0 it return segmentation fault

thohemp commented 2 years ago

Pip package is fixed, thanks https://github.com/thohemp/6DRepNet/commit/cf23e1228379a4123f2ca54b8a37e71fe231fc83

GlennCGL commented 2 years ago

And demo can not run successfully: python SixDRepNet/demo.py --snapshot 6DRepNet_300W_LP_AFLW2000.pth --cam 0 it return segmentation fault

it seems happen in saved_state_dict = torch.load(os.path.join(snapshot_path), map_location='cpu'). Is there something wrong with the ckpt file?

GlennCGL commented 2 years ago

I have fixed problem above. Nice work and demo!