Giter Club home page Giter Club logo

embedding_gan's Introduction

Embedding_GAN

Embedding Adversarial Learning for Vehicle Re-Identification This code is for the work: ******

Note that: the complete code for training and testing will be opened in 2019.

We provide the embedding network without the adversarial learning, which provide the baseline method in our work. the code can be download from https://github.com/yanbai1993/Embedding-Network.

Requirements

  • pytorch
  • Linux
  • Python 2 or 3
  • CPU or NVIDIA GPU + CUDA CuDNN

Getting Started

Installation

  • Install PyTorch and dependencies from http://pytorch.org
  • Install Torch vision from the source.
git clone https://github.com/pytorch/vision
cd vision
python setup.py install
pip install visdom
pip install dominate
  • Clone this repo:
git clone https://github.com/yanbai1993/Embedding_GAN.git
cd Embedding GAN

visualization for hard negative

Based on Embedding GAN, you can generate hard negative samples. To better illustrate the hard negatives generation procedure, we provided the test code and several models (six models under different training stages). The models can be downloaded from "https://pan.baidu.com/s/1vkmccegB5epCa48C7pANbQ". You need to put the models into 'checkpoints' dir.

sh test.sh

The generated images are saved in results dir.

performance test

VehicleID dataset

We provide the test code and models for VehicleID dataset, the model can be download from https://pan.baidu.com/s/1-l6d_VuQ0oXsx6W7PT0xMA. You can test the mAP and CMC result.

sh map.sh
sh cmc.sh

VeRI-776 dataset

For VeRI-776 dataset, we provide an evaluate script, "baseline_evaluation_FACT_776.m", on VeRi following https://github.com/VehicleReId/VeRidataset. You can download our dist file from https://pan.baidu.com/s/1SI9QfviZllwMzqVQbkhh9A. The training code and models will be released later.

embedding_gan's People

Contributors

anorange0409 avatar yanbai1993 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.