No module named 'trt_pose'

imranmu commented 4 years ago

I have successfully installed and build all packages (I hope) but running the example from Jupyter notebook gives error ----> 2 import trt_pose.coco 3 4 with open('human_pose.json', 'r') as f: 5 human_pose = json.load(f) ModuleNotFoundError: No module named 'trt_pose'
I am running it from virtual environment (venvJetssonPoseEstimation) user@user-desktop:~/workspaceJetson/poseEstimation/trt_pose$
These are packages Cython (0.29.16) numpy (1.18.2) Pillow (7.1.1) pip (9.0.1) setuptools (46.1.3) six (1.14.0) torch (1.3.0) torch2trt (0.0.3) torchvision (0.4.2) trt-pose (0.0.1) wheel (0.34.2)

Have also tried with torch v1.4.0 and torchvision v0.5.0 but now revert back to torch v 1.3.0 and torchvision v0.4.2

I could't see any available solution and tried some related problem's solution.

polya-xue commented 4 years ago

same problem

jsynnott commented 4 years ago

Anyone have any success with this? Same problem.

jasonakon commented 4 years ago

please perform sudo python setup.py install and restart your system and try again

anselanza commented 4 years ago

Got the same error, worked for me after sudo python setup.py install from within the repo root.

NVIDIA-AI-IOT / trt_pose