Skip to content

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

Pytorch implementation of "Fast Training of Triplet-based Deep Binary Embedding Networks".

License

Notifications You must be signed in to change notification settings

xwzy/Triplet-deep-hash-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

34 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Triplet-deep-hash-pytorch

Pytorch implementation of "Fast Training of Triplet-based Deep Binary Embedding Networks". http://arxiv.org/abs/1603.02844

Feel free to contribute code.

Update 2017.11.13

Refactor this project.

Use code in https://github.com/kentsommer/keras-inceptionV4 to extract feature.

DEMO

Deep hash for "A", "B".

TODO

  • Add multiclass support.
  • Make code clean.
  • Add more base networks.
  • Add query code for new project.

Usage

Train

  1. Put training pictures in train/[category-id], test pictures in data/test.
  2. Run src/extract_feature/batch_extarct_test.py and src/extract_feature/batch_extract_train.py to extract feature for future use.
  3. Run src/hash_net/generate_random_dataset.py to generate random training data.
  4. Run src/hash_net/hashNet.py to train your triplet deep hash network.

## Test

1. Create folder test, and create pos, neg in test with pictures that you want to retrive.

2. Run testQue.py to query your picture set.

About

Pytorch implementation of "Fast Training of Triplet-based Deep Binary Embedding Networks".

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

Morty Proxy This is a proxified and sanitized view of the page, visit original site.