Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
ltf
 
 
 
 
 
 

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.

About

No description, website, or topics provided.

Resources

Releases

No releases published

Packages

No packages published
You can’t perform that action at this time.