mv resnet50mid to reid-specific model list

pull/119/head
KaiyangZhou 2018-11-09 16:17:29 +00:00
parent fa1ba75eb0
commit 0545358b6f
1 changed files with 1 additions and 1 deletions

View File

@ -24,7 +24,6 @@ __model_factory = {
# image classification models
'resnet50': resnet50,
'resnet50_fc512': resnet50_fc512,
'resnet50mid': resnet50mid,
'resnext50_32x4d': resnext50_32x4d,
'resnext101_32x4d': resnext101_32x4d,
'se_resnet50': se_resnet50,
@ -46,6 +45,7 @@ __model_factory = {
'squeezenet1_1': squeezenet1_1,
# reid-specific models
'mudeep': MuDeep,
'resnet50mid': resnet50mid,
'hacnn': HACNN,
'pcb_p6': pcb_p6,
'pcb_p4': pcb_p4,