nemo_rl.data.hf_datasets.clevr

Contenu du module

Nom	Description
`CLEVRCoGenTDataset`	Aucun

Nom	Description
`pil_to_base64`	Convertit un objet PIL Image en une chaîne encodée en base64.
`format_answer_fromtags`	Extrait le contenu entre les balises <answer> et supprime les espaces.
`format_clevr_cogent_dataset`	Formate le jeu de données CLEVR-CoGenT en un journal de messages de type API OpenAI.
`prepare_clevr_cogent_dataset`	Aucun

nemo_rl.data.hf_datasets.clevr.pil_to_base64(
    image: PIL.Image.Image, format: str = 'PNG'
) -> str

Convertit un objet PIL Image en une chaîne encodée en base64.

Args: image: L’objet PIL Image à convertir. format: Le format de l’image (par exemple, “PNG”, “JPEG”). Par défaut “PNG”.

Returns: Une représentation de l’image en chaîne encodée base64.

nemo_rl.data.hf_datasets.clevr.format_answer_fromtags(answer: str) -> str

Extrait le contenu entre les balises <answer> et supprime les espaces.

nemo_rl.data.hf_datasets.clevr.format_clevr_cogent_dataset(
    example: dict[str, typing.Any], return_pil: bool = False
) -> dict[str, typing.Any]

Formate le jeu de données CLEVR-CoGenT en un journal de messages de type API OpenAI.

nemo_rl.data.hf_datasets.clevr.prepare_clevr_cogent_dataset(
    split: str = 'trainA', task_name: typing.Optional[str] = None
)

class nemo_rl.data.hf_datasets.clevr.CLEVRCoGenTDataset(split: str = 'trainA', prompt_file: typing.Optional[str] = None)