henry123-boy/SpaTracker

Missing code for DPTHeadEnc

dmitryshurov opened this issue · 0 comments

Hi,

Thanks for the release!

I'm trying to run the demo and getting the following issues.

When installing, I had to clone 2 extra repos (co-tracker and Depth-Anything) and copy their contents to the models folder because parts of the code related to these models seemed missing.

git clone https://github.com/henry123-boy/SpaTracker.git
git clone https://github.com/facebookresearch/co-tracker.git
git clone https://github.com/LiheYoung/Depth-Anything.git

cp -Rf co-tracker/cotracker/* SpaTracker/models/cotracker
cp -Rf Depth-Anything/depth_anything/* SpaTracker/models/monoD/depth_anything

After doing that, I'm still facing errors some missing the DPTHeadEnc, which I was not able to find anywhere in the repos:

Traceback (most recent call last):
  File "/content/SpaTracker/demo.py", line 24, in <module>
    from models.spatracker.predictor import SpaTrackerPredictor
  File "/content/SpaTracker/models/spatracker/predictor.py", line 12, in <module>
    from models.spatracker.models.core.spatracker.spatracker import get_points_on_a_grid
  File "/content/SpaTracker/models/spatracker/models/core/spatracker/spatracker.py", line 12, in <module>
    from models.spatracker.models.core.spatracker.blocks import Lie
  File "/content/SpaTracker/models/spatracker/models/core/spatracker/blocks.py", line 20, in <module>
    from models.monoD.depth_anything.dpt import DPTHeadEnc, DPTHead
ImportError: cannot import name 'DPTHeadEnc' from 'models.monoD.depth_anything.dpt'

Am I doing something wrong, or is it a part of the code that was not submitted?

Thank you.