update path

zhedahe · Nov 27, 2020 · d7a27e3 · d7a27e3
1 parent 877b855
commit d7a27e3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tensorflow_classification/Test2_alexnet/trainGPU.py b/tensorflow_classification/Test2_alexnet/trainGPU.py
@@ -37,7 +37,7 @@ def main():
     epochs = 10
 
     # class dict
-    data_class = [cla for cla in os.listdir(train_dir) if os.path.isdir(os.path.join(data_root, cla))]
+    data_class = [cla for cla in os.listdir(train_dir) if os.path.isdir(os.path.join(train_dir, cla))]
     class_num = len(data_class)
     class_dict = dict((value, index) for index, value in enumerate(data_class))
 

diff --git a/tensorflow_classification/Test3_vgg/trainGPU.py b/tensorflow_classification/Test3_vgg/trainGPU.py
@@ -37,7 +37,7 @@ def main():
     epochs = 10
 
     # class dict
-    data_class = [cla for cla in os.listdir(train_dir) if os.path.isdir(os.path.join(data_root, cla))]
+    data_class = [cla for cla in os.listdir(train_dir) if os.path.isdir(os.path.join(train_dir, cla))]
     class_num = len(data_class)
     class_dict = dict((value, index) for index, value in enumerate(data_class))
 

diff --git a/tensorflow_classification/Test4_goolenet/trainGPU.py b/tensorflow_classification/Test4_goolenet/trainGPU.py
@@ -37,7 +37,7 @@ def main():
     epochs = 30
 
     # class dict
-    data_class = [cla for cla in os.listdir(train_dir) if os.path.isdir(os.path.join(data_root, cla))]
+    data_class = [cla for cla in os.listdir(train_dir) if os.path.isdir(os.path.join(train_dir, cla))]
     class_num = len(data_class)
     class_dict = dict((value, index) for index, value in enumerate(data_class))