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

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +6 -6
utils.py CHANGED
@@ -34,12 +34,12 @@ base_model = tf.keras.applications.ResNet50V2(include_top=False, input_shape=inp
34
  #Making the layers of the model trainable
35
  base_model.trainable = True
36
 
37
- def preprocess(img_path):
38
- img = Image.open(img_path)
39
- img = img.resize((256, 256))
40
- img_array = np.array(img)
41
- img_array = np.expand_dims(img_array, axis=0)
42
- return img_array
43
 
44
  def model_arc():
45
  model = tf.keras.Sequential([
 
34
  #Making the layers of the model trainable
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
+
42
+ return image
43
 
44
  def model_arc():
45
  model = tf.keras.Sequential([