Skip to content

Commit

Permalink
Initial commit
Browse files Browse the repository at this point in the history
  • Loading branch information
zydou committed Jul 18, 2018
0 parents commit ba99993
Show file tree
Hide file tree
Showing 39 changed files with 2,726 additions and 0 deletions.
112 changes: 112 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,112 @@
# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
*$py.class

# C extensions
*.so

# Distribution / packaging
.Python
env/
build/
develop-eggs/
dist/
downloads/
eggs/
.eggs/
lib/
lib64/
parts/
sdist/
var/
wheels/
*.egg-info/
.installed.cfg
*.egg

# PyInstaller
# Usually these files are written by a python script from a template
# before PyInstaller builds the exe, so as to inject date/other infos into it.
*.manifest
*.spec

# Installer logs
pip-log.txt
pip-delete-this-directory.txt

# Unit test / coverage reports
htmlcov/
.tox/
.coverage
.coverage.*
.cache
nosetests.xml
coverage.xml
*.cover
.hypothesis/

# Translations
*.mo
*.pot

# Django stuff:
*.log
local_settings.py

# Flask stuff:
instance/
.webassets-cache

# Scrapy stuff:
.scrapy

# Sphinx documentation
docs/_build/

# PyBuilder
target/

# Jupyter Notebook
.ipynb_checkpoints

# pyenv
.python-version

# celery beat schedule file
celerybeat-schedule

# SageMath parsed files
*.sage.py

# dotenv
.env

# virtualenv
.venv
venv/
ENV/

# Spyder project settings
.spyderproject
.spyproject

# Rope project settings
.ropeproject

# mkdocs documentation
/site

# mypy
.mypy_cache/

# protobuf
*_pb2.py

# experiments logs
experiments/
log*/
eval*/
.vscode/
._.DS_Store
*.dylib
65 changes: 65 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,65 @@
# Deep-Person: Learning Discriminative Deep Features for Person Re-Identification

Code for [Deep-Person: Learning Discriminative Deep Features for Person Re-Identification](https://arxiv.org/abs/1711.10658) based on [open-reid](https://github.com/Cysu/open-reid).

## Prerequisites
Deep-Person is developed and tested with Pytorch 0.2 and Python 3.6.

[Anaconda](https://www.anaconda.com/download/) is required to finish the Installation.
## Installation

```
cd DeepPerson/
conda env create -f environment.yml
```

This will create an environment named "deepperson". (use `conda list` to see all environments)

## Datasets Preparation
[Market1501](http://www.liangzheng.org/Project/project_reid.html)

[DukeMTMC-Reid](https://github.com/layumi/DukeMTMC-reID_evaluation)

[CUHK03](https://docs.google.com/spreadsheet/viewform?usp=drive_web&formkey=dHRkMkFVSUFvbTJIRkRDLWRwZWpONnc6MA#gid=0)


## Training and evaluation
**NOTE: You must activate "deepperson" environment first before running the code.**

To activate "deepperson" environment:

```
conda activate deepperson
```
To train a model:

```
cd DeepPerson/
python examples/deep.py -d market1501 --logs-dir logs/market
```

To evaluate a pretrained model:

```
cd DeepPerson/
python examples/deep.py -d market1501 --resume logs/market/checkpoint.pth.tar --evaluate
```

We provide a pretrained model on Market1501 which can be found at our [release page](https://github.com/zydou/DeepPerson/releases).
## Citation

If you find this project helpful for your research, please cite the following paper:

```
@article{xbai2017deepperson,
author = {Xiang Bai and
Mingkun Yang and
Tengteng Huang and
Zhiyong Dou and
Rui Yu and
Yongchao Xu},
title = {Deep-Person: Learning Discriminative Deep Features for Person Re-Identification},
journal = {arXiv preprint arXiv:1711.10658},
year = {2017},
}
```
50 changes: 50 additions & 0 deletions environment.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
name: deepperson
channels:
- soumith
- defaults
dependencies:
- ca-certificates=2018.03.07=0
- certifi=2018.4.16=py36_0
- cffi=1.11.5=py36h9745a5d_0
- freetype=2.8=hab7d2ae_1
- h5py=2.7.1=py36h3585f63_0
- hdf5=1.10.1=h9caa474_1
- intel-openmp=2018.0.0=8
- jpeg=9b=h024ee3a_2
- libedit=3.1=heed3624_0
- libffi=3.2.1=hd88cf55_4
- libgcc=7.2.0=h69d50b8_2
- libgcc-ng=7.2.0=hdf63c60_3
- libgfortran-ng=7.2.0=hdf63c60_3
- libpng=1.6.34=hb9fc6fc_0
- libstdcxx-ng=7.2.0=hdf63c60_3
- libtiff=4.0.9=h28f6b97_0
- mkl=2018.0.2=1
- mkl_fft=1.0.1=py36h3010b51_0
- mkl_random=1.0.1=py36h629b387_0
- ncurses=6.0=h9df7e31_2
- numpy=1.14.2=py36hdbf6ddf_1
- olefile=0.45.1=py36_0
- openssl=1.0.2o=h20670df_0
- pillow=5.1.0=py36h3deb7b8_0
- pip=10.0.1=py36_0
- pycparser=2.18=py36hf9f622e_1
- python=3.6.5=hc3d631a_2
- readline=7.0=ha6073c6_4
- scikit-learn=0.19.1=py36h7aa7ec6_0
- scipy=1.0.1=py36hfc37229_0
- setuptools=39.1.0=py36_0
- six=1.11.0=py36h372c433_1
- sqlite=3.23.1=he433501_0
- tk=8.6.7=hc745277_3
- wheel=0.31.0=py36_0
- xz=5.2.3=h5e939de_4
- zlib=1.2.11=ha838bed_2
- cuda80=1.0=0
- pytorch=0.2.0=py36h53baedd_4cu80
- torchvision=0.1.9=py36h7584368_1
- pip:
- metric-learn==0.3.0
- torch==0.2.0.post4
prefix: /home/zydou/anaconda3/envs/deepperson

Loading

0 comments on commit ba99993

Please sign in to comment.