examples: More updates and fixes

Author: George Bisbas

examples/seismic/tutorials/08_snapshotting.ipynb

@@ -25,7 +25,6 @@
    "outputs": [],
    "source": [
     "#NBVAL_IGNORE_OUTPUT\n",
-    "%reset -f\n",
     "import numpy as np\n",
     "import matplotlib.pyplot as plt\n",
     "%matplotlib inline"
@@ -90,7 +89,6 @@
     "# This cell sets up the problem that is already explained in the first TLE tutorial.\n",
     "\n",
     "#NBVAL_IGNORE_OUTPUT\n",
-    "#%%flake8\n",
     "from devito import TimeFunction, Eq, solve, Operator\n",
     "\n",
     "from examples.seismic import Receiver\n",

examples/seismic/tutorials/13_LSRTM_acoustic.ipynb

@@ -454,40 +454,33 @@
    ],
    "source": [
     "#NBVAL_IGNORE_OUTPUT\n",
-    "image_up_dev = np.zeros((model0.vp.shape[0], model0.vp.shape[1]),dtype)\n",
+    "image_up_dev = np.zeros((model0.vp.shape[0], model0.vp.shape[1]), dtype)\n",
     "\n",
     "image = np.zeros((model0.vp.shape[0], model0.vp.shape[1]))\n",
     "\n",
     "nrec = 101\n",
     "niter = 20  # Number of iterations of the LSRTM\n",
     "history = np.zeros((niter, 1))  # Objective function\n",
     "\n",
-    "image_prev = np.zeros((model0.vp.shape[0],model0.vp.shape[1]))\n",
+    "image_prev = np.zeros((model0.vp.shape[0], model0.vp.shape[1]))\n",
     "\n",
-    "grad_prev  = np.zeros((model0.vp.shape[0],model0.vp.shape[1]))\n",
+    "grad_prev  = np.zeros((model0.vp.shape[0], model0.vp.shape[1]))\n",
     "\n",
-    "yk  = np.zeros((model0.vp.shape[0],model0.vp.shape[1]))\n",
+    "yk  = np.zeros((model0.vp.shape[0], model0.vp.shape[1]))\n",
     "\n",
-    "sk = np.zeros((model0.vp.shape[0],model0.vp.shape[1]))\n",
+    "sk = np.zeros((model0.vp.shape[0], model0.vp.shape[1]))\n",
     "\n",
     "for k in range(niter):\n",
+    "    print('LSRTM Iteration', k+1)\n",
     "\n",
-    "    dm =  image_up_dev  # Reflectivity for Calculated data via Born\n",
-    "\n",
-    "    print('LSRTM Iteration',k+1)\n",
-    "\n",
-    "    objective,grad_full,d_obs,d_syn = lsrtm_gradient(dm)\n",
-    "\n",
+    "    # Feed the reflectivity model to the LSRTM function\n",
+    "    objective, grad_full, d_obs, d_syn = lsrtm_gradient(image_up_dev)\n",
     "    history[k] = objective\n",
-    "\n",
     "    yk = grad_full.data - grad_prev\n",
-    "\n",
     "    sk = image_up_dev - image_prev\n",
-    "\n",
     "    alfa = get_alfa(yk, sk, k)\n",
     "\n",
     "    grad_prev = grad_full.data\n",
-    "\n",
     "    image_prev = image_up_dev\n",
     "\n",
     "    image_up_dev = image_up_dev - alfa*grad_full.data\n",
@@ -654,7 +647,7 @@
    ],
    "source": [
     "#NBVAL_IGNORE_OUTPUT\n",
-    "slices=tuple(slice(model.nbl,-model.nbl) for _ in range(2))\n",
+    "slices = tuple(slice(model.nbl, -model.nbl) for _ in range(2))\n",
     "dm_true = (solver.model.vp.data**(-2) - model0.vp.data**(-2))[slices]\n",
     "plot_image(np.diff(dm_true, axis=1))"
    ]
@@ -681,9 +674,9 @@
     "#NBVAL_SKIP\n",
     "plt.figure(figsize=(8, 9))\n",
     "x = np.linspace(0, 1, 101)\n",
-    "plt.plot(rtm[50,:], x, color=plt.gray(), linewidth=2)\n",
-    "plt.plot(lsrtm[50,:],x, 'r',linewidth=2)\n",
-    "plt.plot(dm_true[50,:],x, 'k--',linewidth=2)\n",
+    "plt.plot(rtm[50, :], x, color=plt.gray(), linewidth=2)\n",
+    "plt.plot(lsrtm[50, :],x, 'r', linewidth=2)\n",
+    "plt.plot(dm_true[50, :],x, 'k--', linewidth=2)\n",
     "\n",
     "plt.legend(['Initial reflectivity', 'Reflectivity via LSRTM','True Reflectivity'], fontsize=15)\n",
     "plt.ylabel('Depth (Km)')\n",

examples/seismic/tutorials/15_tti_qp_pure.ipynb

@@ -77,11 +77,11 @@
     "model = demo_model('layers-tti', spacing=spacing, space_order=8,\n",
     "                   shape=shape, nbl=nbl, nlayers=1)\n",
     "\n",
-    "# initialize Thomsem parameters to those used in Mu et al., (2020)\n",
-    "model.update('vp', np.ones(shape)*3.6)  # km/s\n",
-    "model.update('epsilon', np.ones(shape)*0.23)\n",
-    "model.update('delta', np.ones(shape)*0.17)\n",
-    "model.update('theta', np.ones(shape)*(45.*(np.pi/180.)))  # radians"
+    "# Initialize Thomsem parameters to those used in Mu et al., (2020)\n",
+    "model.update('vp', np.full(shape, 3.6))  # km/s\n",
+    "model.update('epsilon', np.full(shape, 0.23))\n",
+    "model.update('delta', np.full(shape, 0.17))\n",
+    "model.update('theta', np.full(shape, (45.*(np.pi/180.))))  # radians"
    ]
   },
   {