Skip to content
/ CBR Public
forked from hueifang/CBR

Huei-Fang Yang, Kevin Lin, Ting-Yen Chen, and Chu-Song Chen, "Cross-batch Reference Learning for Deep Retrieval," IEEE Transactions on Neural Networks and Learning Systems, 2019

Notifications You must be signed in to change notification settings

ivclab/CBR

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Cross-batch Reference Learning for Deep Classification and Retrieval

by Huei-Fang Yang, Kevin Lin, and Chu-Song Chen

Introduction

This code implements the cross-batch reference (CBR) learning as described in our ACM MM 2016 paper.

Citing the CBR

If you find CBR useful in your research, please consider citing:

@inproceedings{yang:acmmm16,
    author    = {Huei-Fang Yang and
                 Kevin Lin and
                 Chu-Song Chen},
    title     = {Cross-batch Reference Learning for Deep Classification and Retrieval},
    booktitle = {Proc. ACM MM},
    pages     = {1237--1246},
    year      = {2016}
}

Requirements

  1. Caffe, matcaffe and pycaffe (see: Caffe installation instructions)
  2. MATLAB (required for performance evaluation)

Train a network with CBR on CIFAR-10

  1. Download the pretrained model and CIFAR-10 dataset.

    $./download.sh
    
  2. Modify the CAFFE_BIN in run.sh to the path where the CAFFE is installed.

  3. Modify the caffe_root in update_model.py to the path where the CAFFE is installed.

  4. Launch the script to train a network with CBR. This would take a few hours.

    $./run.sh
    

Evaluate the trained model

  1. Modify the addpath in run_cifar10.m to the path where the CAFFE is installed.

  2. Launch MATLAB and run the evaluation code to obtain mAP and precition@k.

    >>run_cifar10
    

Contact

Please feel free to contact Huei-Fang Yang (hfyang@citi.sinica.edu.tw), Kevin Lin (kevinlin311.tw@iis.sinica.edu.tw), or Chu-Song Chen (song@iis.sinica.edu.tw) if you had any questions.

About

Huei-Fang Yang, Kevin Lin, Ting-Yen Chen, and Chu-Song Chen, "Cross-batch Reference Learning for Deep Retrieval," IEEE Transactions on Neural Networks and Learning Systems, 2019

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • MATLAB 50.0%
  • Python 30.3%
  • Shell 19.7%