apple / ml-depth-pro

Depth Pro: Sharp Monocular Metric Depth in Less Than a Second.
Other
3.73k stars 250 forks source link

RuntimeError: operator torchvision::nms does not exist #62

Open EngrAwab opened 3 weeks ago

EngrAwab commented 3 weeks ago

Hey I ran depth-pro-run -i ./data/example.jpg and got the following command Traceback (most recent call last): File "/home/saith/miniconda3/envs/depth-pro/bin/depth-pro-run", line 5, in from depth_pro.cli import run_main File "/home/saith/Desktop/data/ml-depth-pro/src/depth_pro/init.py", line 4, in from .depth_pro import create_model_and_transforms # noqa File "/home/saith/Desktop/data/ml-depth-pro/src/depth_pro/depth_pro.py", line 12, in from torchvision.transforms import ( File "/home/saith/miniconda3/envs/depth-pro/lib/python3.9/site-packages/torchvision/init.py", line 10, in from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip File "/home/saith/miniconda3/envs/depth-pro/lib/python3.9/site-packages/torchvision/_meta_registrations.py", line 164, in def meta_nms(dets, scores, iou_threshold): File "/home/saith/miniconda3/envs/depth-pro/lib/python3.9/site-packages/torch/library.py", line 795, in register use_lib._register_fake(op_name, func, _stacklevel=stacklevel + 1) File "/home/saith/miniconda3/envs/depth-pro/lib/python3.9/site-packages/torch/library.py", line 184, in _register_fake handle = entry.fake_impl.register(func_to_register, source) File "/home/saith/miniconda3/envs/depth-pro/lib/python3.9/site-packages/torch/_library/fake_impl.py", line 31, in register if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, "Meta"): RuntimeError: operator torchvision::nms does not exist