We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 97d1172 commit 2f726d1Copy full SHA for 2f726d1
data_set/split_data.py
@@ -21,7 +21,8 @@ def main():
21
cwd = os.getcwd()
22
data_root = os.path.join(cwd, "flower_data")
23
origin_flower_path = os.path.join(data_root, "flower_photos")
24
- assert os.path.exists(origin_flower_path)
+ assert os.path.exists(origin_flower_path), "path '{}' does not exist.".format(origin_flower_path)
25
+
26
flower_class = [cla for cla in os.listdir(origin_flower_path)
27
if os.path.isdir(os.path.join(origin_flower_path, cla))]
28
0 commit comments