import glob
import numpy as np
import os
-import Queue
+import queue
import random
import scipy
from scipy import misc
# Start prefetching thread
# self.prefetch_queue = Queue.Queue(maxsize=prefetch_size)
- self.prefetch_queue = Queue.PriorityQueue(maxsize=prefetch_size)
+ self.prefetch_queue = queue.PriorityQueue(maxsize=prefetch_size)
for k in range(num_workers):
t = threading.Thread(target=prefetch_job,
args=(self.load_fn, self.prefetch_queue, self.data_list,
# load_fn_frame3 = partial(dataset_frame3.process_func)
# p_queue_frame3 = PrefetchQueue(load_fn_frame3, data_list_frame3, FLAGS.batch_size, shuffle=False, num_workers=num_workers)
- for step in xrange(0, FLAGS.max_steps):
+ for step in range(0, FLAGS.max_steps):
batch_idx = step % epoch_num
batch_data_list_frame1 = data_list_frame1[int(batch_idx * FLAGS.batch_size) : int((batch_idx + 1) * FLAGS.batch_size)]