diff --git a/latent_feature.ipynb b/latent_feature.ipynb
index 2ed1f9f9..4e1ffe35 100644
--- a/latent_feature.ipynb
+++ b/latent_feature.ipynb
@@ -98,9 +98,9 @@
     "        img = Image.open(os.path.join(data_path, i))\n",
     "        img = img.resize((224, 224))\n",
     "        img = np.array(img) / 255.\n",
-    "        img[...,0] = (img[...,0] - img[...,0].mean())/img[...,0].std()\n",
-    "        img[...,1] = (img[...,1] - img[...,1].mean())/img[...,1].std()\n",
-    "        img[...,2] = (img[...,2] - img[...,2].mean())/img[...,2].std()\n",
+    "        imagenet_mean = np.array([0.485, 0.456, 0.406])\n",
+    "        imagenet_std  = np.array([0.229, 0.224, 0.225])\n",
+    "        img = (img - imagenet_mean) / imagenet_std\n",
     "        assert img.shape == (224, 224, 3)\n",
     "        \n",
     "        latent_feature = run_one_image(img, model_,arch)\n",
@@ -193,4 +193,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}
\ No newline at end of file