OutputIterator

class lib.infer.iterator.OutputIterator(queue: Queue[QueueItemInT | ExtractSignal], name: str, plugin_type: T.Literal['detect', 'align', 'mask', 'identity', 'file'], batch_size: int, error_state: ErrorState)

Bases: ExtractIterator[ExtractBatch, FrameFaces]

Handles parsing incoming ExtractBatch objects into FrameFaces objects and yielding one frame at a time from the pipeline

Parameters:

  • queue (Queue[QueueItemInT | ExtractSignal]) – The output queue from the plugin runner

  • name (str) – The plugin name and process calling this iterator

  • plugin_type (T.Literal['detect', 'align', 'mask', 'identity', 'file']) – The type of extractor plugin that this iterator is serving

  • batch_size (int) – The batch size that data should be returned from the iterator

  • error_state (ErrorState)

__next__() → FrameFaces

Get the next batch of data from the iterator

Return type:

A FrameFaces object for a single frame

Raises:

StopIteration – When the input is exhausted