From 6c87b92c76931e0b0aa4e2af135812112306e89b Mon Sep 17 00:00:00 2001 From: Evan Shelhamer Date: Mon, 28 Jul 2014 20:10:16 -0700 Subject: [PATCH] [example] add caffe to pythonpath in all examples --- examples/imagenet_classification.ipynb | 7 +++++-- examples/net_surgery.ipynb | 7 ++++++- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/examples/imagenet_classification.ipynb b/examples/imagenet_classification.ipynb index e7fe9f8..40e69ae 100644 --- a/examples/imagenet_classification.ipynb +++ b/examples/imagenet_classification.ipynb @@ -31,9 +31,12 @@ "import matplotlib.pyplot as plt\n", "%matplotlib inline\n", "\n", - "import caffe\n", - "\n", + "# Make sure that caffe is on the python path:\n", "caffe_root = '../' # this file is expected to be in {caffe_root}/examples\n", + "import sys\n", + "sys.path.insert(0, caffe_root + 'python')\n", + "\n", + "import caffe\n", "\n", "# Set the right path to your model definition file, pretrained model weights,\n", "# and the image you would like to classify.\n", diff --git a/examples/net_surgery.ipynb b/examples/net_surgery.ipynb index 0a6e208..d64ae85 100644 --- a/examples/net_surgery.ipynb +++ b/examples/net_surgery.ipynb @@ -135,6 +135,11 @@ "cell_type": "code", "collapsed": false, "input": [ + "# Make sure that caffe is on the python path:\n", + "caffe_root = '../' # this file is expected to be in {caffe_root}/examples\n", + "import sys\n", + "sys.path.insert(0, caffe_root + 'python')\n", + "\n", "import caffe\n", "\n", "# Load the original network and extract the fully-connected layers' parameters.\n", @@ -326,4 +331,4 @@ "metadata": {} } ] -} +} \ No newline at end of file -- 2.7.4