Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
caffe2.python.record_queue.RecordQueue Class Reference
Inheritance diagram for caffe2.python.record_queue.RecordQueue:

Public Member Functions

def __init__ (self, fields, name=None, capacity=1, enforce_unique_name=False, num_threads=1)
 
def build (self, reader, process=None)
 

Public Attributes

 schema
 
 name
 
 num_threads
 
 blobs_queue
 
 writer
 
 reader
 
 exit_step
 

Detailed Description

The class is used to feed data with some process from a reader into a
    queue and provider a reader interface for data fetching from the queue.

Definition at line 48 of file record_queue.py.

Member Function Documentation

def caffe2.python.record_queue.RecordQueue.build (   self,
  reader,
  process = None 
)
Build the producer_step to feed data from reader into the queue, and
return the reader interface.
Inputs:
    reader:           read data which will be stored in the queue.
    process:          preprocess data before enqueue.
Outputs:
    reader:           reader to fetch the data from the queue.
    producer_step:    the step insert the data into the queue. Should be
              run with comsume_step together.
    exit_step:        the step to close queue
    schema:           the schema for the reader.

Definition at line 80 of file record_queue.py.


The documentation for this class was generated from the following file: