BoxLoader.py 文件源码

def __init__(self, sources=[], initOnStart=True):
        self.totalCount = 0
        self.counts=[]
        self.sources=[]
        self.initDone=False
        self.initOnStart=initOnStart

        with tf.name_scope('dataset') as scope:
            self.queue = tf.FIFOQueue(dtypes=[tf.float32, tf.float32, tf.uint8],
                    capacity=self.QUEUE_CAPACITY)

            self.image = tf.placeholder(dtype=tf.float32, shape=[None, None, 3], name="image")
            self.boxes = tf.placeholder(dtype=tf.float32, shape=[None,4], name="boxes")
            self.classes = tf.placeholder(dtype=tf.uint8, shape=[None], name="classes")

            self.enqueueOp = self.queue.enqueue([self.image, self.boxes, self.classes])

        self.sources=sources[:]