├── README.md ├── figs └── pipeline.png ├── reid ├── criterion │ ├── __init__.py │ ├── build_criterion.py │ ├── softmax_loss.py │ ├── spatial_align_loss.py │ └── triplet_loss.py ├── data │ ├── __init__.py │ ├── bases.py │ ├── build_data.py │ ├── market1501.py │ ├── msmt17.py │ ├── preprocessing.py │ ├── sampler.py │ ├── vehicleid.py │ └── veri.py ├── engine │ ├── __init__.py │ ├── evaluator.py │ ├── searcher.py │ └── trainer.py ├── models │ ├── __init__.py │ ├── cm.py │ ├── msinet.py │ ├── operations.py │ ├── sam.py │ └── search_cnn.py ├── solver │ ├── __init__.py │ ├── build_optimizer.py │ └── lr_scheduler.py └── utils │ ├── __init__.py │ ├── logging.py │ ├── meters.py │ ├── metrics.py │ ├── osutils.py │ ├── rerank.py │ └── serialization.py ├── search.py └── train.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/README.md -------------------------------------------------------------------------------- /figs/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/figs/pipeline.png -------------------------------------------------------------------------------- /reid/criterion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/criterion/__init__.py -------------------------------------------------------------------------------- /reid/criterion/build_criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/criterion/build_criterion.py -------------------------------------------------------------------------------- /reid/criterion/softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/criterion/softmax_loss.py -------------------------------------------------------------------------------- /reid/criterion/spatial_align_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/criterion/spatial_align_loss.py -------------------------------------------------------------------------------- /reid/criterion/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/criterion/triplet_loss.py -------------------------------------------------------------------------------- /reid/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/data/__init__.py -------------------------------------------------------------------------------- /reid/data/bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/data/bases.py -------------------------------------------------------------------------------- /reid/data/build_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/data/build_data.py -------------------------------------------------------------------------------- /reid/data/market1501.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/data/market1501.py -------------------------------------------------------------------------------- /reid/data/msmt17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/data/msmt17.py -------------------------------------------------------------------------------- /reid/data/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/data/preprocessing.py -------------------------------------------------------------------------------- /reid/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/data/sampler.py -------------------------------------------------------------------------------- /reid/data/vehicleid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/data/vehicleid.py -------------------------------------------------------------------------------- /reid/data/veri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/data/veri.py -------------------------------------------------------------------------------- /reid/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/engine/__init__.py -------------------------------------------------------------------------------- /reid/engine/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/engine/evaluator.py -------------------------------------------------------------------------------- /reid/engine/searcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/engine/searcher.py -------------------------------------------------------------------------------- /reid/engine/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/engine/trainer.py -------------------------------------------------------------------------------- /reid/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/models/__init__.py -------------------------------------------------------------------------------- /reid/models/cm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/models/cm.py -------------------------------------------------------------------------------- /reid/models/msinet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/models/msinet.py -------------------------------------------------------------------------------- /reid/models/operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/models/operations.py -------------------------------------------------------------------------------- /reid/models/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/models/sam.py -------------------------------------------------------------------------------- /reid/models/search_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/models/search_cnn.py -------------------------------------------------------------------------------- /reid/solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/solver/__init__.py -------------------------------------------------------------------------------- /reid/solver/build_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/solver/build_optimizer.py -------------------------------------------------------------------------------- /reid/solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/solver/lr_scheduler.py -------------------------------------------------------------------------------- /reid/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reid/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/utils/logging.py -------------------------------------------------------------------------------- /reid/utils/meters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/utils/meters.py -------------------------------------------------------------------------------- /reid/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/utils/metrics.py -------------------------------------------------------------------------------- /reid/utils/osutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/utils/osutils.py -------------------------------------------------------------------------------- /reid/utils/rerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/utils/rerank.py -------------------------------------------------------------------------------- /reid/utils/serialization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/reid/utils/serialization.py -------------------------------------------------------------------------------- /search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/search.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vimar-gu/MSINet/HEAD/train.py --------------------------------------------------------------------------------