diff --git a/examples/ImageNet Pretrained Network (VGG_S).ipynb b/examples/ImageNet Pretrained Network (VGG_S).ipynb index 38b8489..87b383d 100644 --- a/examples/ImageNet Pretrained Network (VGG_S).ipynb +++ b/examples/ImageNet Pretrained Network (VGG_S).ipynb @@ -127,7 +127,10 @@ "source": [ "import pickle\n", "\n", - "model = pickle.load(open('vgg_cnn_s.pkl'))\n", + "try:\n", + " values = pickle.load(open('vgg19_normalized.pkl','rb'))['param values']\n", + "except UnicodeDecodeError:\n", + " values = pickle.load(open('vgg19_normalized.pkl','rb'),encoding='Latin-1')['param values'] # python3\n", "CLASSES = model['synset words']\n", "MEAN_IMAGE = model['mean image']\n", "\n",