nemo_curator.stages.image.io.convert

Module Contents

Classes

Name	Description
`ConvertImageBatchToDocumentBatchStage`	Convert image batch to DocumentBatch

API

class nemo_curator.stages.image.io.convert.ConvertImageBatchToDocumentBatchStage(
    fields: list[str] = list(),
    name: str = 'convert_image_batch_to_doc...
)

Dataclass

Bases: ProcessingStage[ImageBatch, DocumentBatch]

Convert image batch to DocumentBatch

Parameters:

fields

list[str]Defaults to list()

list of fields of ImageObject to convert to DocumentBatch

fields

list[str] = field(default_factory=list)

name

str = 'convert_image_batch_to_document_batch'

nemo_curator.stages.image.io.convert.ConvertImageBatchToDocumentBatchStage.process(
    task: nemo_curator.tasks.ImageBatch
) -> nemo_curator.tasks.DocumentBatch

Convert image batch to DocumentBatch