There is a newer version of the record available.

Published November 12, 2018 | Version v0.5.0

KaiyangZhou/deep-person-reid: deep-person-reid v0.5.0

Authors/Creators

  • 1. QMUL
  • 2. Zhejiang University

Description

Major updates:

  • Model codes such as resnet.py and densenet.py keep the original style for easier modification.
  • Generalize CrossEntropyLableSmooth to CrossEntropyLoss. --label-smooth should be called explicitly in order to add the label smoothing regularizer to the cross entropy loss.
  • Add support to multi-dataset training. Datasets are specified by the arguments -s and -t, which refer to source datasets and target datasets, respectively. Both can take multiple strings delimited by space. For example, say you wanna train a model using Market1501+DukeMTMC-reID, just set -s market1501 dukemtmcreid. If you wanna test on multiple datasets, you can do -t market1501 dukemtmcreid cuhk03 msmt17.
  • Arguments are unified in args.py.
  • Dataloaders are wrapped into two classes, which are ImageDataManager and VideoDataManager (see data_manager.py). A datamanger is initialized by dm = ImageDataManager(use_gpu, **image_dataset_kwargs(args)) where image_dataset_kwargs() is implemented in args.py. Therefore, when new arguments are added to the data manager, you don't need to exhausively change everywhere in the code. What you need to update are (1) add new arguments in args.py and (2) update the input arguments in data_manager.py.
  • BENCHMARK is replaced with MODEL_ZOO where model weights and training scripts can be downloaded.

Files

KaiyangZhou/deep-person-reid-v0.5.0.zip

Files (304.4 kB)

Name Size Download all
md5:e788905e3ebb08c5a744d6836af117c5
304.4 kB Preview Download

Additional details