MODEL_ZOO.md

August 2, 2020 ยท View on GitHub

Model Zoo

For the leaderboard on public benchmarks, please refer to LEADERBOARD.md.

Note: all the models below are selected by the performance on validation sets.

Unsupervised learning (USL) on object re-ID

  • Direct infer models are directly tested on the re-ID datasets with ImageNet pre-trained weights.

Market-1501

MethodBackbonePre-trainedmAP(%)top-1(%)top-5(%)top-10(%)Train timeDownload
Direct inferResNet50ImageNet2.26.714.920.1n/a
UDA_TPResNet50ImageNet34.758.674.078.9~2h[config] [model]
strong_baselineResNet50ImageNet70.587.995.797.1~2.5h[config] [model]
MMTResNet50ImageNet74.388.196.097.5~4.5h[config] [model]
SpCLResNet50ImageNet76.089.596.297.5~2h[config] [model]

DukeMTMC-reID

MethodBackbonePre-trainedmAP(%)top-1(%)top-5(%)top-10(%)Train timeDownload
Direct inferResNet50ImageNet2.37.514.718.1n/a
UDA_TPResNet50ImageNet42.364.476.079.9~2h[config] [model]
strong_baselineResNet50ImageNet54.772.983.587.2~2.5h[config] [model]
MMTResNet50ImageNet60.375.686.089.2~4.5h[config] [model]
SpCLResNet50ImageNet67.182.490.893.0~2h[config] [model]

... (TBD)

Unsupervised domain adaptation (UDA) on object re-ID

  • Direct infer models are trained on the source-domain datasets (source_pretrain) and directly tested on the target-domain datasets.
  • UDA methods (MMT, SpCL, etc.) starting from ImageNet means that they are trained end-to-end in only one stage without source-domain pre-training.

DukeMTMC-reID -> Market-1501

MethodBackbonePre-trainedmAP(%)top-1(%)top-5(%)top-10(%)Train timeDownload
Direct inferResNet50DukeMTMC27.258.975.781.4~1h[config] [model]
UDA_TPResNet50DukeMTMC52.376.087.891.9~2h[config] [model]
strong_baselineResNet50ImageNet75.690.996.697.8~3h[config] [model]
MMTResNet50ImageNet80.992.297.698.4~6h[config] [model]
SpCLResNet50ImageNet78.290.596.697.8~3h[config] [model]

Market-1501 -> DukeMTMC-reID

MethodBackbonePre-trainedmAP(%)top-1(%)top-5(%)top-10(%)Train timeDownload
Direct inferResNet50Market28.149.364.369.7~1h[config] [model]
UDA_TPResNet50Market45.765.578.081.7~2h[config] [model]
strong_baselineResNet50ImageNet60.475.986.289.8~3h[config] [model]
MMTResNet50ImageNet67.780.389.992.9~6h[config] [model]
SpCLResNet50ImageNet70.483.891.293.4~3h[config] [model]

... (TBD)