-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
0 parents
commit ba99993
Showing
39 changed files
with
2,726 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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}, | ||
} | ||
``` |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 | ||
|
Oops, something went wrong.