Skip to content

P
python3-imagej-tiff
Commit 54dda3d1 authored by Oleg Dzhimiev's avatar Oleg Dzhimiev
Browse files
Options

1. sort tfrecord lists 

2. update tfrecord file name pattern
parent 9ceb0063
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
nn_ds_inmem4_tmp.py
View file @ 54dda3d1
...@@ -122,11 +122,12 @@ except IndexError: ...@@ -122,11 +122,12 @@ except IndexError:
# if the path is a directory # if the path is a directory
if os.path.isdir(train_filenameTFR): if os.path.isdir(train_filenameTFR):
train_filesTFR = glob.glob(train_filenameTFR+"/*train*.tfrecords") train_filesTFR = glob.glob(train_filenameTFR+"/*train-*.tfrecords")
train_filenameTFR = train_filesTFR[0] train_filenameTFR = train_filesTFR[0]
else: else:
train_filesTFR = [train_filenameTFR] train_filesTFR = [train_filenameTFR]
train_filesTFR.sort()
print("Train tfrecords: "+str(train_filesTFR)) print("Train tfrecords: "+str(train_filesTFR))
# tfrecords' paths for testing # tfrecords' paths for testing
...@@ -142,6 +143,7 @@ if os.path.isdir(test_filenameTFR): ...@@ -142,6 +143,7 @@ if os.path.isdir(test_filenameTFR):
else: else:
test_filesTFR = [test_filenameTFR] test_filesTFR = [test_filenameTFR]
test_filesTFR.sort()
print("Test tfrecords: "+str(test_filesTFR)) print("Test tfrecords: "+str(test_filesTFR))
# Now we are left with 2 lists - train and test list # Now we are left with 2 lists - train and test list
......
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