Skip to content

Integrate EfficientTAM #640

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.

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
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

cjaverliat
Copy link

Integrate EfficientTAM

This PR integrates the EfficientTAM model from the paper "Efficient Track Anything". The components originate from the repository of the authors and the config file were slightly adapted for being usable in this repository.

"Efficient Track Anything"
Yunyang Xiong, Chong Zhou, Xiaoyu Xiang, Lemeng Wu, Chenchen Zhu, Zechun Liu, Saksham Suri, Balakrishnan Varadarajan, Ramya Akula, Forrest Iandola, Raghuraman Krishnamoorthi, Bilge Soran, Vikas Chandra
arXiv:2411.18933 (Paper Link)

Usage

Download the weights by running

cd checkpoints
./download_ckpts_efficienttam.sh

The models can then be loaded as any other models (the video predictor notebook was updated for an example):

from sam2.build_sam import build_sam2_video_predictor

sam2_checkpoint = "../checkpoints/efficienttam_s_1.pt"
model_cfg = "configs/efficienttam/efficienttam_s_1.yaml"

predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint, device=device)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants