search

facebookresearch / co-tracker

CoTracker is a model for tracking any point (pixel) on a video.
https://co-tracker.github.io/
Other
2.52k stars 177 forks source link

Running on CPU #9

Closed shersoni610 closed 11 months ago

shersoni610 commented 11 months ago

Hello,

How can I run this code on MAC-M1 or cpu? It seems cuda is hardcorded somewhere.

Thanks

nikitakaraevv commented 11 months ago

Hi @shersoni610, I've resolved this issue today. Please clone the repo again and let me know if it works. Also, now there's an easy way to try the model through torch.hub:

import torch
import timm
import einops
import tqdm

cotracker = torch.hub.load("facebookresearch/co-tracker", "cotracker_w8")
shersoni610 commented 11 months ago

Thank you very much. I will try again and give you feedback.