From a73e483a0dcf3257d40174b9a69e6c5d488a997c Mon Sep 17 00:00:00 2001
From: parallelArchitect <236280545+parallelArchitect@users.noreply.github.com>
Date: Thu, 25 Jun 2026 04:49:52 -0400
Subject: [PATCH] fix: use ImageNet normalization in latent_feature.ipynb (#53)

---
 latent_feature.ipynb | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

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