Merge pull request #902 from zlheui/update-largedataset_cnn-example
Update dataset path for the largedataset_cnn example
diff --git a/examples/largedataset_cnn/process_data.py b/examples/largedataset_cnn/process_data.py
index 7d92d01..8489268 100644
--- a/examples/largedataset_cnn/process_data.py
+++ b/examples/largedataset_cnn/process_data.py
@@ -35,7 +35,6 @@
return im
-
def process_data(dataset_root, classes):
# load class names
with open(classes, 'r', encoding='utf-8') as f:
@@ -79,7 +78,7 @@
return train_input_paths, train_labels, val_input_paths, val_labels
def loaddata():
- dataset_root = '/Dataset/Data/'
+ dataset_root = '/Dataset/Data/' # need to set this path an argument
classes = '/Dataset/classes.txt'
return process_data(dataset_root, classes)