Skip to content

Commit 2f726d1

Browse files
author
wz
committed
add assert info
1 parent 97d1172 commit 2f726d1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

data_set/split_data.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,8 @@ def main():
2121
cwd = os.getcwd()
2222
data_root = os.path.join(cwd, "flower_data")
2323
origin_flower_path = os.path.join(data_root, "flower_photos")
24-
assert os.path.exists(origin_flower_path)
24+
assert os.path.exists(origin_flower_path), "path '{}' does not exist.".format(origin_flower_path)
25+
2526
flower_class = [cla for cla in os.listdir(origin_flower_path)
2627
if os.path.isdir(os.path.join(origin_flower_path, cla))]
2728

0 commit comments

Comments
 (0)