Skip to content
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

Add a Hugging Face Spaces Demo for SAMURAI #6

Open
Itayflymingo opened this issue Nov 22, 2024 · 4 comments
Open

Add a Hugging Face Spaces Demo for SAMURAI #6

Itayflymingo opened this issue Nov 22, 2024 · 4 comments
Labels
enhancement (no plan yet) New feature or request

Comments

@Itayflymingo
Copy link

This repository provides an excellent implementation of SAMURAI: Adapting Segment Anything Model for Zero-Shot Visual Tracking with Motion-Aware Memory. To make the model more accessible and user-friendly, we propose creating an interactive demo on Hugging Face Spaces.

The demo should allow users to upload their custom videos or image sequences, provide a bounding box for the first frame, and visualize the tracking results. This interactive experience will greatly enhance the visibility and usability of SAMURAI for researchers and practitioners.

Requirements for the Demo:

Frontend:

A simple interface with file upload options for videos or frame directories.
A text box to input the bounding box of the first frame in xywh format.
Backend:

Use the existing SAMURAI scripts for inference (demo.py).
Automatically process the uploaded files and return the tracking results (e.g., video or annotated frames).
Environment Setup:

Use Gradio or similar frameworks supported by Hugging Face Spaces.
Ensure the required dependencies (torch, torchvision, opencv-python, etc.) are installed.
Output:

Display the tracking results in the web interface (e.g., overlay bounding boxes on the video).
Option to download the processed output.
References:

SAMURAI Repository
SAM 2 Repository
Hugging Face Spaces Documentation
Tasks:

Set up a Hugging Face Space and deploy the demo using the SAMURAI implementation.
Test the demo with custom videos and frames to ensure reliability.
Document the setup and usage of the demo in the repository's README.
Expected Outcome:
An interactive demo hosted on Hugging Face Spaces that enables users to experiment with SAMURAI for zero-shot visual tracking effortlessly.

@yangchris11 yangchris11 added the enhancement (no plan yet) New feature or request label Nov 23, 2024
@MarsX-2002
Copy link

is this request still open for contribution?

@ashwath007
Copy link

@MarsX-2002 yes i guess

@Itayflymingo
Copy link
Author

go for it @MarsX-2002

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement (no plan yet) New feature or request
Projects
None yet
Development

No branches or pull requests

5 participants
@yangchris11 @ashwath007 @MarsX-2002 @Itayflymingo and others