Cysu / open-reid

Open source person re-identification library in python
https://cysu.github.io/open-reid/
MIT License
1.34k stars 349 forks source link

Refactor for v0.2.0 #7

Closed Cysu closed 7 years ago

Cysu commented 7 years ago

Three main changes:

  1. Unify the interface for datasets and models. Now one can call reid.models.create('resnet50', ...) and reid.datasets.create('cuhk03', ...) to create instances.

  2. Reorganize examples by their methods rather than the net architecture, for example, softmax loss classification, triplet loss verification, and OIM loss. Make the dataset and net architecture command line parameters for the examples.

  3. Now saving / loading checkpoints gets rid of DataParallel in examples.