Skip to content

Improve discoverability by pushing Co-Tracker to HF along with download stats #99

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
NielsRogge wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Improve discoverability by pushing Co-Tracker to HF along with download stats #99

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
153b6a3
Add first draft
NielsRogge Jul 19, 2024
91d2f80
Add metadata
NielsRogge Jul 19, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion cotracker/models/core/cotracker/cotracker.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@
import torch.nn as nn
import torch.nn.functional as F

from huggingface_hub import PyTorchModelHubMixin

from cotracker.models.core.model_utils import sample_features4d, sample_features5d
from cotracker.models.core.embeddings import (
get_2d_embedding,
Expand All @@ -26,7 +28,11 @@
torch.manual_seed(0)


class CoTracker2(nn.Module):
class CoTracker2(nn.Module, PyTorchModelHubMixin,
library_name="co-tracker",
repo_url="https://github.com/facebookresearch/co-tracker",
license="cc-by-nc-4.0",
tags=["object-tracking"]):
def __init__(
self,
window_len=8,
Expand Down