[example] add caffe to pythonpath in all examples

diff --git a/examples/imagenet_classification.ipynb b/examples/imagenet_classification.ipynb
index e7fe9f8..40e69ae 100644 (file)
--- 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 (file)
--- 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