Commit 1d988652 authored by Oleg Dzhimiev's avatar Oleg Dzhimiev
Browse files

multiple files

parent e544dc0c

nn_ds_inmem4_tmp.py

+22 −2
Original line number Diff line number Diff line
@@ -112,15 +112,33 @@ def read_and_decode(filename_queue): 
#http://adventuresinmachinelearning.com/introduction-tensorflow-queuing/



# Main code



# tfrecords' paths for training

try:

    train_filenameTFR =  sys.argv[1]

except IndexError:

    train_filenameTFR = "/mnt/dde6f983-d149-435e-b4a2-88749245cc6c/home/eyesis/x3d_data/data_sets/tf_data/train.tfrecords"



# if the path is a directory

if os.path.isdir(train_filenameTFR):

    train_filesTFR = glob.glob(train_filenameTFR+"/train_*.tfrecords")

    train_filenameTFR = train_filesTFR[0] 

else:

    train_filesTFR = [train_filenameTFR]

     

# tfrecords' paths for testing    

try:

    test_filenameTFR =  sys.argv[2]

except IndexError:

    test_filenameTFR = "/mnt/dde6f983-d149-435e-b4a2-88749245cc6c/home/eyesis/x3d_data/data_sets/tf_data/test.tfrecords"

#FILES_PER_SCENE



# if the path is a directory

if os.path.isdir(test_filenameTFR):

    test_filesTFR = glob.glob(test_filenameTFR+"/test_*.tfrecords")

    test_filenameTFR = test_filesTFR[0] 

else:

    test_filesTFR = [test_filenameTFR]

    



import tensorflow as tf

import tensorflow.contrib.slim as slim
@@ -379,6 +397,8 @@ def read_new_tfrecord_file(filename,result): 


tfrecord_filename = train_filenameTFR



tfrecord_file_counter = 0 



with tf.Session()  as sess:

    

    sess.run(tf.global_variables_initializer())