Skip to content

ffmpbgrnn/tflibs

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 

Repository files navigation

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.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published