diff --git a/models/Inception.py b/models/Inception.py index 160bd81..aac1174 100644 --- a/models/Inception.py +++ b/models/Inception.py @@ -9,6 +9,10 @@ import sys __all__ = ['InceptionV4ReID'] +""" +Code imported from https://github.com/Cadene/pretrained-models.pytorch +""" + pretrained_settings = { 'inceptionv4': { 'imagenet': {