VishnuEcoClim commited on
Commit
047c523
·
1 Parent(s): 3f6eaf3

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -8,7 +8,7 @@ import numpy as np
8
  from PIL import Image
9
 
10
  def gen_labels():
11
- train = 'Dataset/Train'
12
  train_generator = ImageDataGenerator(rescale=1/255)
13
 
14
  train_generator = train_generator.flow_from_directory(train,
 
8
  from PIL import Image
9
 
10
  def gen_labels():
11
+ train = 'Dataset'
12
  train_generator = ImageDataGenerator(rescale=1/255)
13
 
14
  train_generator = train_generator.flow_from_directory(train,