search

Daniil-Osokin / lightweight-human-pose-estimation-3d-demo.pytorch

Real-time 3D multi-person pose estimation demo in PyTorch. OpenVINO backend can be used for fast inference on CPU.
Apache License 2.0
653 stars 137 forks source link

Cannot convert to .onnx #88

Closed adammpolak closed 2 years ago

adammpolak commented 2 years ago

Running:

Traceback (most recent call last):
  File "scripts/convert_to_onnx.py", line 5, in <module>
    from models.with_mobilenet import PoseEstimationWithMobileNet
ModuleNotFoundError: No module named 'models'

causes errors.

Is there a reason I am not able to convert the .pth to .onnx using the included script?

adammpolak commented 2 years ago

I pulled from the OpenVINO model zoo and converted using OpenVINO utilities, no longer necessary

Daniil-Osokin commented 2 years ago

Great!