No description, website, or topics provided.
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
ltf
.gitignore
README.md

README.md

This repo contains source code for

  • Bidirectional Multirate Reconstruction for Temporal Modeling in Videos, Linchao Zhu, Zhongwen Xu, Yi Yang.
  • Few-Shot Object Recognition from Machine-Labeled Web Images, Zhongwen Xu*, Linchao Zhu*, Yi Yang

To run our key-value memory experiments, you need to install TensorFlow 1.0 or above. The source code has been tested on TensorFlow 1.3.

# Download files (dataset, features, etc) for training
wget http://libstorage.pub/Expr/Mem/mem.zip -O /tmp/mem.zip
cd /tmp && unzip mem.zip

# Go to the source directory
# To train the model on GPU 0, run
./mem.sh train 0
# It will read the configure file configs/mem/few_shot_100.py

# To evaludate the model, run
./mem.sh eval /tmp/mem/few_shots/train_dir/{RUN_ID}/model.ckpt-${MODEL_ID} 1

Change the `train_stage` variable to `train_5` or `train_10` to evaluate the model on 5-shot and 10-shot settings.
You can obtain almost the same performance as reported on our paper.