Python threaded_input_pipelineの例

コード例 #1

ファイル: train.py プロジェクト: ezglory/cnn_lstm_ctc_ocr

def _get_single_input():
    """Set up and return image, label, and width tensors"""

    image, width, label, length, text, filename = mjsynth.threaded_input_pipeline(
        deps.get('records'),
        str.split(FLAGS.filename_pattern, ','),
        batch_size=1,
        num_threads=FLAGS.num_input_threads,
        num_epochs=1,
        batch_device=FLAGS.input_device,
        preprocess_device=FLAGS.input_device)
    return image, width, label, length, text, filename

コード例 #2

ファイル: train.py プロジェクト: trigrass2/cnn_lstm_ctc_ocr

def _get_single_input():
    """Set up and return image, label, and width tensors"""

    image,width,label,length,text,filename=mjsynth.threaded_input_pipeline(
        deps.get('records'), 
        str.split(FLAGS.filename_pattern,','),
        batch_size=1,
        num_threads=FLAGS.num_input_threads,
        num_epochs=1,
        batch_device=FLAGS.input_device, 
        preprocess_device=FLAGS.input_device )
    return image,width,label,length,text,filename

コード例 #3

ファイル: test.py プロジェクト: schperics/cnn_lstm_ctc_ocr

def _get_input():
    """Set up and return image, label, width and text tensors"""

    image, width, label, length, text, filename = mjsynth.threaded_input_pipeline(
        FLAGS.test_path,
        str.split(FLAGS.filename_pattern, ','),
        batch_size=FLAGS.batch_size,
        num_threads=FLAGS.num_input_threads,
        num_epochs=None,  # Repeat for streaming
        batch_device=FLAGS.device,
        preprocess_device=FLAGS.device)

    return image, width, label, length

コード例 #4

ファイル: test.py プロジェクト: trigrass2/cnn_lstm_ctc_ocr

def _get_input():
    """Set up and return image, label, width and text tensors"""

    image,width,label,length,text,filename=mjsynth.threaded_input_pipeline(
        FLAGS.test_path,
        str.split(FLAGS.filename_pattern,','),
        batch_size=FLAGS.batch_size,
        num_threads=FLAGS.num_input_threads,
        num_epochs=None, # Repeat for streaming
        batch_device=FLAGS.device, 
        preprocess_device=FLAGS.device )
    
    return image,width,label,length