Commit ce6fccbe authored by Sebastian Brodehl's avatar Sebastian Brodehl
Browse files

Catch error if directory already exists, and continue script.

parent 28b91429
...@@ -49,6 +49,7 @@ from __future__ import division ...@@ -49,6 +49,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import os import os
import errno
import os.path import os.path
import sys import sys
...@@ -69,7 +70,13 @@ if __name__ == '__main__': ...@@ -69,7 +70,13 @@ if __name__ == '__main__':
# Make all sub-directories in the validation data dir. # Make all sub-directories in the validation data dir.
for label in unique_labels: for label in unique_labels:
labeled_data_dir = os.path.join(data_dir, label) labeled_data_dir = os.path.join(data_dir, label)
# Catch error if sub-directory exists
try:
os.makedirs(labeled_data_dir) os.makedirs(labeled_data_dir)
except OSError as e:
# Raise all errors but 'EEXIST'
if e.errno != errno.EEXIST:
raise
# Move all of the image to the appropriate sub-directory. # Move all of the image to the appropriate sub-directory.
for i in range(len(labels)): for i in range(len(labels)):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment