stages.image.io.convert#

Module Contents#

Classes#

ConvertImageBatchToDocumentBatchStage

Convert image batch to DocumentBatch

API#

class stages.image.io.convert.ConvertImageBatchToDocumentBatchStage#

Bases: nemo_curator.stages.base.ProcessingStage[nemo_curator.tasks.ImageBatch, nemo_curator.tasks.DocumentBatch]

Convert image batch to DocumentBatch

Args: fields: list of fields of ImageObject to convert to DocumentBatch

fields: list[str]#

‘field(…)’

process(
task: nemo_curator.tasks.ImageBatch,
) nemo_curator.tasks.DocumentBatch#

Convert image batch to DocumentBatch