From 97665de85959e2c71894eb9b28acff755e657eba Mon Sep 17 00:00:00 2001
From: Jianping Zhang <91776033+jpzhang1810@users.noreply.github.com>
Date: Tue, 18 Jul 2023 16:47:13 +0800
Subject: [PATCH] Update dataset.py

---
 dataset.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/dataset.py b/dataset.py
index a7e56ae..4c8f9b3 100644
--- a/dataset.py
+++ b/dataset.py
@@ -18,10 +18,10 @@ DEFAULT_CROP_PCT = 0.875 # 0.9, 1.0
 IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)
 IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
 # Imagenet 21k
-#IMAGENET_INCEPTION_MEAN = (0.5, 0.5, 0.5)
-#IMAGENET_INCEPTION_STD = (0.5, 0.5, 0.5)
-IMAGENET_INCEPTION_MEAN = (0.0, 0.0, 0.0)
-IMAGENET_INCEPTION_STD = (1.0, 1.0, 1.0)
+IMAGENET_INCEPTION_MEAN = (0.5, 0.5, 0.5)
+IMAGENET_INCEPTION_STD = (0.5, 0.5, 0.5)
+#IMAGENET_INCEPTION_MEAN = (0.0, 0.0, 0.0)
+#IMAGENET_INCEPTION_STD = (1.0, 1.0, 1.0)
 
 
 def params(model_name):
@@ -117,4 +117,4 @@ class CNNDataset(data.Dataset):
         img = transforms.Resize((299,299))(img)
         img = transforms.Compose([transforms.ToTensor()])(img)
         #print(img.shape)
-        return img, class_id, class_name, image_name
\ No newline at end of file
+        return img, class_id, class_name, image_name