Change image size

CihanSoylu · Apr 2, 2020 · 84d2e5b · 84d2e5b
1 parent 77db41c
commit 84d2e5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/project_utils/project_5_utils.py b/project_utils/project_5_utils.py
@@ -32,7 +32,7 @@ def load_cats_dogs():
     dataset = tfds.load('cats_vs_dogs', split = 'train', shuffle_files=True)
 
     def resize(item):
-        return tf.image.resize(item['image'], (224,224)), item['label']
+        return tf.image.resize(item['image'], (160,160)), item['label']
 
     dataset = dataset.map(resize)