Update HAPs (more like LFs)

Notebooks/HAPs.ipynb

@@ -30,18 +30,21 @@
    "outputs": [],
    "source": [
     "# Specific Utils\n",
-    "im_size = 2\n",
     "\n",
     "def DFC_filter(raster):\n",
     "    raster[raster > 1e4] = raster[raster < 1e4].max()\n",
     "\n",
-    "def show(im):\n",
+    "def show(im, im_size=1, save=None):\n",
     "    plt.figure(figsize=(16*im_size,3*im_size))\n",
     "    plt.imshow(im)\n",
     "    plt.colorbar()\n",
+    "    \n",
+    "    if save is not None:\n",
+    "        plt.savefig(save, bbox_inches='tight', pad_inches=1)\n",
+    "    \n",
     "    plt.show()\n",
     "\n",
-    "def mshow(Xs, titles=None):\n",
+    "def mshow(Xs, titles=None, im_size=1, save=None):\n",
     "    s = len(Xs)\n",
     "\n",
     "    plt.figure(figsize=(16*im_size,3*im_size*s))\n",
@@ -54,8 +57,10 @@
     "            plt.title(titles[i])\n",
     "            \n",
     "        plt.colorbar()\n",
+    "    \n",
+    "    if save is not None:\n",
+    "        plt.savefig(save, bbox_inches='tight', pad_inches=1)\n",
     "        \n",
-    "    plt.savefig('test.png', bbox_inches='tight', pad_inches=1)\n",
     "    plt.show()"
    ]
   },
@@ -128,7 +133,7 @@
    "outputs": [],
    "source": [
     "a = attributes\n",
-    "i = 7\n",
+    "i = 3\n",
     "show(a[:,:,i].astype(np.float) - a[:,:,i+1])"
    ]
   },
@@ -145,29 +150,28 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "patch_size = 5\n",
+    "def create_patches(array, patch_size=3):\n",
+    "    amp = int((patch_size - 1 ) / 2)\n",
     "\n",
-    "### \n",
+    "    stack = list()\n",
-    "amp = int((patch_size - 1 ) / 2)\n",
+    "    for i in range(-amp, amp+1):\n",
+    "        ai = i if i > 0 else None\n",
+    "        bi = i if i < 0 else None\n",
+    "        ci = -bi if bi is not None else None\n",
+    "        di = -ai if ai is not None else None\n",
     "\n",
-    "stack = list()\n",
+    "        for j in range(-amp, amp+1):\n",
-    "for i in range(-amp, amp+1):\n",
+    "            offset = np.zeros(array.shape)\n",
-    "    ai = i if i > 0 else None\n",
+    "            aj = j if j > 0 else None\n",
-    "    bi = i if i < 0 else None\n",
+    "            bj = j if j < 0 else None\n",
-    "    ci = -bi if bi is not None else None\n",
+    "            cj = -bj if bj is not None else None\n",
-    "    di = -ai if ai is not None else None\n",
+    "            dj = -aj if aj is not None else None\n",
+    "            print('{}:{} {}:{} - {}:{} {}:{}'.format(ai, bi, ci, di, aj, bj, cj, dj))\n",
+    "            offset[ai:bi, aj:bj] = array[ci:di, cj:dj]\n",
+    "            stack.append(offset)\n",
+    "    return np.stack(stack, axis=-1)\n",
     "\n",
-    "    for j in range(-amp, amp+1):\n",
+    "offset_stack = create_patches(attributes, 5)\n",
-    "        offset = np.zeros(raster.shape)\n",
-    "        aj = j if j > 0 else None\n",
-    "        bj = j if j < 0 else None\n",
-    "        cj = -bj if bj is not None else None\n",
-    "        dj = -aj if aj is not None else None\n",
-    "        print('{}:{} {}:{} - {}:{} {}:{}'.format(ai, bi, ci, di, aj, bj, cj, dj))\n",
-    "        offset[ai:bi, aj:bj] = raster[ci:di, cj:dj]\n",
-    "        stack.append(offset)\n",
-    "\n",
-    "offset_stack = np.stack(stack, axis=2)\n",
     "offset_stack.shape"
    ]
   },
@@ -190,6 +194,24 @@
     "stack_std.shape"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "figs = list()\n",
+    "ttls = list()\n",
+    "for d in range(stack_std.shape[-1]):\n",
+    "    if d == 0:\n",
+    "        ttls.append('Origin')\n",
+    "    else:\n",
+    "        ttls.append('STD 5x5 from area {}'.format(area[d-1]))\n",
+    "    figs.append(stack_std[:,:,d])\n",
+    "    \n",
+    "mshow(figs, ttls, 2)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -228,6 +250,32 @@
     "       ]\n",
     "mshow(figs, ttls)"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "size = '{}x{}'.format(patch_size, patch_size)\n",
+    "figs = [raster,\n",
+    "        stack_avr[:,:,-1],\n",
+    "        stack_mean[:,:,-1],\n",
+    "        stack_min[:,:,-1],\n",
+    "        stack_max[:,:,-1],\n",
+    "        stack_var[:,:,-1],\n",
+    "        stack_std[:,:,-1]\n",
+    "       ]\n",
+    "ttls = ['Origin',\n",
+    "        'Average ' + size,\n",
+    "        'Mean ' + size,\n",
+    "        'Minimum ' + size,\n",
+    "        'Maximum ' + size,\n",
+    "        'Variance ' + size,\n",
+    "        'STD ' + size\n",
+    "       ]\n",
+    "mshow(figs, ttls, 2, save='mstack2.png')"
+   ]
   }
  ],
  "metadata": {