def __init__(self, path, batch_size=16, input_size=227,
scale_factor=1.0, num_threads=10):
self._path = path
self._list_files = glob.glob(os.path.join(path, "**/*.avi"))
self._batch_size = batch_size
self._scale_factor = scale_factor
self._image_size = input_size
self._label_size = int(input_size * self._scale_factor)
self._num_threads = num_threads
self._coord = tf.train.Coordinator()
self._image_shape = [batch_size, self._image_size, self._image_size, 3]
self._label_shape = [batch_size, self._label_size, self._label_size, 1]
p_x = tf.placeholder(tf.float32, self._image_shape, name='x')
p_y = tf.placeholder(tf.float32, self._label_shape, name='y')
inputs = [p_x, p_y]
self._queue = tf.FIFOQueue(400,
[i.dtype for i in inputs], [i.get_shape() for i in inputs])
self._inputs = inputs
self._enqueue_op = self._queue.enqueue(inputs)
self._queue_close_op = self._queue.close(cancel_pending_enqueues=True)
self._threads = []
video_avi_flow_saliency.py 文件源码
python
阅读 29
收藏 0
点赞 0
评论 0
评论列表
文章目录