Pip install failed

[glennccc]

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]

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

[thohemp]

Pip package is fixed, thanks cf23e12

[GlennCGL]

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]

I have fixed problem above.
Nice work and demo!