mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update model

pull/17/head
KaiyangZhou 2018-05-04 23:14:21 +01:00
parent cd3a565ad0
commit 91c33c7719
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/InceptionResNetV2.py
View File

@ -7,7 +7,7 @@ import torch.utils.model_zoo as model_zoo
import os import os
import sys import sys
__all__ = ['InceptionResNetV2', 'inceptionresnetv2'] __all__ = ['InceptionResNetV2']
pretrained_settings = { pretrained_settings = {
'inceptionresnetv2': { 'inceptionresnetv2': {

2
models/__init__.py
View File

@ -12,6 +12,7 @@ from .InceptionV4 import *
from .SEResNet import * from .SEResNet import *
from .NASNet import * from .NASNet import *
from .DPN import * from .DPN import *
from .InceptionResNetV2 import *
__factory = { __factory = {
'resnet50': ResNet50, 'resnet50': ResNet50,
@ -27,6 +28,7 @@ __factory = {
'seresnet50': SEResNet50, 'seresnet50': SEResNet50,
'nasnet': NASNetAMobile, 'nasnet': NASNetAMobile,
'dpn92': DPN, 'dpn92': DPN,
'inceptionresnetv2': InceptionResNetV2,
} }
def get_names(): def get_names():