Skip to content
GraphBit: Bitwise Interaction Mining via Deep Reinforcement Learning
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
README.md Update README.md Jun 21, 2018
RL_network.py Add files via upload Jun 16, 2018
US_network.py Add files via upload Jun 16, 2018
bifeat_extract.m Add files via upload Jun 16, 2018
main.py Add files via upload Jun 16, 2018
precision.m Add files via upload Jun 16, 2018
retrival.m Add files via upload Jun 16, 2018
test-label.txt Add files via upload Jun 16, 2018
train-label.txt Add files via upload Jun 16, 2018

README.md

GraphBit

GraphBit: Bitwise Interaction Mining via Deep Reinforcement Learning This repo contains the basic code for Graphbit on CIFAR-10.

Quick Start

The code is based on Keras and TensorFlow. Please prepare the real-valued feature in 4096 dimensions of CIFAR-10 by downing the data through the link, and run main.py to train the network and get hashed compact features. Finally you can use bifeat_extract.m to extract the binary feature and run retrival.m to get the mAP on retrival task.

Repo organization

The repo is organized as follows:

Citation

If you find Graphbit useful, please cite it.

@inproceedings{duan2018graphbit,
title={GraphBit: Bitwise Interaction Mining via Deep Reinforcement Learning},
author={Duan, Yueqi and Wang, Ziwei and Lu, Jiwen and Lin, Xudong and Zhou, Jie},
booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
pages={8270--8279},
year={2018}
}
You can’t perform that action at this time.