aveek-saha / viraliq

Code implementation & CLI tool for the paper: "Graph Based Temporal Aggregation for Video Retrieval"

Graph Based Temporal Aggregation for Video Retrieval

VIRALIQ: VIdeo RetrievAL through Image Queries

If you're looking for just the source code, it can be found on the source branch.

Dependencies

FFmpeg

FFmpeg is required for extracting frames from videos. Download links can be found here.

Note: Windows users will have to add the FFmpeg bin folder to the PATH variable.

Installation

Install the package using pip

pip install viraliq

CLI usage

Create embeddings

Create video embeddings

viraliq cluster "PATH_TO_VIDEO_DIR"

Search

Search using an image query

viraliq search "PATH_TO_VIDEO_DIR" "PATH_TO_IMAGE"

Citing

If you find VIRALIQ useful please cite the following paper:

Srinivasan, Arvind, Aprameya Bharadwaj, Aveek Saha, and Subramanyam Natarajan. "Graph Based Temporal Aggregation for Video Retrieval." arXiv preprint arXiv:2011.02426 (2020).