Refactor data loader

titu1994 · Jan 7, 2018 · 0d4ab67 · 0d4ab67
1 parent 1c8d655
commit 0d4ab67
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 4 deletions.
diff --git a/extract_nasnet_features.py b/extract_nasnet_features.py
@@ -2,7 +2,7 @@
 from keras import backend as K
 
 from utils.nasnet import NASNetMobile
-from data_loader import train_generator, val_generator
+from utils.data_loader import train_generator, val_generator
 
 sess = tf.Session()
 K.set_session(sess)

diff --git a/pretrain_nasnet_mobile.py b/pretrain_nasnet_mobile.py
@@ -6,7 +6,7 @@
 from keras.optimizers import Adam
 from keras import backend as K
 
-from data_loader import features_generator
+from utils.data_loader import features_generator
 
 '''
 Below is a modification to the TensorBoard callback to perform 

diff --git a/train_mobilenet.py b/train_mobilenet.py
@@ -7,7 +7,7 @@
 from keras.optimizers import Adam
 from keras import backend as K
 
-from data_loader import train_generator, val_generator
+from utils.data_loader import train_generator, val_generator
 
 '''
 Below is a modification to the TensorBoard callback to perform 

diff --git a/train_nasnet_mobile.py b/train_nasnet_mobile.py
@@ -8,7 +8,7 @@
 
 from utils.nasnet import NASNetMobile
 
-from data_loader import train_generator, val_generator
+from utils.data_loader import train_generator, val_generator
 
 '''
 Below is a modification to the TensorBoard callback to perform 

diff --git a/data_loader.py → utils/data_loader.py b/data_loader.py → utils/data_loader.py