Skip to content

yuanmengzhixing/pytorch_deep_metric_learning

 
 

Repository files navigation

Deep Metric Learning

modefied from github.com/bnuwangxun/Deep_Metric

Learn a deep metric which can be used image retrieval , clustering.

============================

Pytorch Code for deep metric methods:

Dataset

  • Car-196

    first 98 classes as train set and last 98 classes as test set

  • CUB-200-2011

    first 98 classes as train set and last 98 classes as test set

  • Stanford-Online

    for the experiments, we split 59,551 images of 11,318 classes for training and 60,502 images of 11,316 classes for testing

    After downloading all the three data file, you should precess them as above, and put the directionary named DataSet in the project. We provide a script to precess CUB( Deep_Metric/DataSet/split_dataset.py ). The other two are similar, you can modify the script by yourself.

Pretrained models in Pytorch

Inceptionn BN network as other metric learning papers do, to save your time, we already download them down and put on my Baidu YunPan.

We also put inception v3 in the Baidu YunPan, the performance of inception v-3 is a little worse(about 1.5% on recall@1 ) than inception BN on CUB/Car datasets.

The download site(https://pan.baidu.com/s/1snmKa1v)

Prerequisites

  • Computer with Linux or OSX
  • PyTorch

NOTE!!!

 To exactly reproduce the result in my paper, please make sure to use the same version of pytorch with me: !!! 0.2.0_3 there are some problem for other version to load the pretrained model of inception-BN.

  • For training, an NVIDIA GPU is strongly recommended for speed. CPU is supported but training may be slow.

Reproducing Car-196 (or CUB-200-2011) experiments

With our loss based on fussy clustering:

sh run_train.sh

To reproduce other experiments, you can edit the run_train.sh file by yourself.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.4%
  • Shell 0.6%