Name		Name	Last commit message	Last commit date
Latest commit History 25 Commits
ActivityNet_process		ActivityNet_process
Charades_STA_process		Charades_STA_process
models		models
Baseline_cross_att.py		Baseline_cross_att.py
README.md		README.md
baseline_train.py		baseline_train.py
baseline_train_fl.py		baseline_train_fl.py
hyper_param.json		hyper_param.json

Repository files navigation

FedVMR

PyTorch implementation of Federated Video Moment Retrieval (FedVMR).

Getting started

Prerequisites

Prepare feature files

Download feature.tar.gz (31GB). After downloading the feature file, extract it to the feature directory under fedvmr directory:

tar zxvf feature.tar.gz

It contains I3D video features of Charades-STA (charades_i3d_rgb.hdf5), C3D video features of ActivityNet-Captions (sub_activitynet_v1-3.c3d.hdf5) and text features (6B.300d.npy).

Install dependencies.

Python 3.7
PyTorch 1.8.0
Cuda 11.1
nltk
tqdm
h5py
easydict

Running Code

python baseline_train_fl.py

About

No description, website, or topics provided.

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%