Commit 9ceb0063 authored by Oleg Dzhimiev's avatar Oleg Dzhimiev

new pattern

parent 2fbab86b
...@@ -42,7 +42,7 @@ SHUFFLE_EPOCH = True ...@@ -42,7 +42,7 @@ SHUFFLE_EPOCH = True
NET_ARCH = 3 # overwrite with argv? NET_ARCH = 3 # overwrite with argv?
#DEBUG_PACK_TILES = True #DEBUG_PACK_TILES = True
SUFFIX=str(NET_ARCH)+ (["R","A"][ABSOLUTE_DISPARITY]) SUFFIX=str(NET_ARCH)+ (["R","A"][ABSOLUTE_DISPARITY])
MAX_TRAIN_FILES_TFR = 1 MAX_TRAIN_FILES_TFR = 4
#http://stackoverflow.com/questions/287871/print-in-terminal-with-colors-using-python #http://stackoverflow.com/questions/287871/print-in-terminal-with-colors-using-python
class bcolors: class bcolors:
HEADER = '\033[95m' HEADER = '\033[95m'
...@@ -122,7 +122,7 @@ except IndexError: ...@@ -122,7 +122,7 @@ 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]
......
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