VishnuEcoClim commited on
Commit
cc5d9a1
·
1 Parent(s): 136c1d6

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -35,7 +35,7 @@ base_model = tf.keras.applications.ResNet50V2(include_top=False, input_shape=inp
35
  base_model.trainable = True
36
 
37
  def preprocess(image):
38
- image = np.array(image.resize((300, 300), Image.ANTIALIAS))
39
  image = np.array(image, dtype='uint8')
40
  image = np.array(image)/255.0
41
 
 
35
  base_model.trainable = True
36
 
37
  def preprocess(image):
38
+ image = np.array(image.resize((256, 256), Image.LANCZOS))
39
  image = np.array(image, dtype='uint8')
40
  image = np.array(image)/255.0
41