diff --git a/models/InceptionResNetV2.py b/models/InceptionResNetV2.py index 3c04da3..ba867ec 100644 --- a/models/InceptionResNetV2.py +++ b/models/InceptionResNetV2.py @@ -7,7 +7,7 @@ import torch.utils.model_zoo as model_zoo import os import sys -__all__ = ['InceptionResNetV2', 'inceptionresnetv2'] +__all__ = ['InceptionResNetV2'] pretrained_settings = { 'inceptionresnetv2': { diff --git a/models/__init__.py b/models/__init__.py index 1641120..8b804b0 100644 --- a/models/__init__.py +++ b/models/__init__.py @@ -12,6 +12,7 @@ from .InceptionV4 import * from .SEResNet import * from .NASNet import * from .DPN import * +from .InceptionResNetV2 import * __factory = { 'resnet50': ResNet50, @@ -27,6 +28,7 @@ __factory = { 'seresnet50': SEResNet50, 'nasnet': NASNetAMobile, 'dpn92': DPN, + 'inceptionresnetv2': InceptionResNetV2, } def get_names():