X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=utils%2Fprefetch_queue_shuffle.py;fp=utils%2Fprefetch_queue_shuffle.py;h=b488649bf01703b7fbc76adf8d75a267ee7de93c;hb=bed00c52fa6fbd9bda76f0f4d2fa2e7c0ff08922;hp=5d391a9bf7e60d2084906581753bf1c2e5844a0a;hpb=c74057bc77fb9bc4eb75668896adfb85ddb99251;p=voxel-flow

diff --git a/utils/prefetch_queue_shuffle.py b/utils/prefetch_queue_shuffle.py
index 5d391a9..b488649 100755
--- a/utils/prefetch_queue_shuffle.py
+++ b/utils/prefetch_queue_shuffle.py
@@ -3,7 +3,7 @@ from __future__ import print_function
 import glob
 import numpy as np
 import os 
-import Queue
+import queue
 import random
 import scipy
 from scipy import misc
@@ -55,7 +55,7 @@ class PrefetchQueue(object):
 
     # 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,