search

NVlabs / FoundationPose

[CVPR 2024 Highlight] FoundationPose: Unified 6D Pose Estimation and Tracking of Novel Objects
https://nvlabs.github.io/FoundationPose/
Other
955 stars 99 forks source link

Error when run run_demo.py? #61

Closed aghaphd closed 3 weeks ago

aghaphd commented 1 month ago

Traceback (most recent call last): File "/home/ali/FoundationPose/run_demo.py", line 52, in pose = est.register(K=reader.K, rgb=color, depth=depth, ob_mask=mask, iteration=args.est_refine_iter) File "/home/ali/FoundationPose/estimater.py", line 173, in register depth = erode_depth(depth, radius=2, device='cuda') NameError: name 'erode_depth' is not defined

Thanks.

wenbowen123 commented 1 month ago

are you running inside a docker? Did you install warp package?

aghaphd commented 1 month ago

Thank you for your response, actually not using the docker. I started the implementation from option 2. Where Warp package mentioned? I mean which step?

wenbowen123 commented 1 month ago

https://github.com/NVlabs/FoundationPose/blob/main/requirements.txt#L79 It's mentioned in option2

aghaphd commented 3 weeks ago

I successfully run it through the Docker.