Diffusers documentation
Результаты
Результаты
Все выходные данные моделей являются подклассами BaseOutput, структурами данных, содержащими всю информацию, возвращаемую моделью. Выходные данные также можно использовать как кортежи или словари.
Например:
from diffusers import DDIMPipeline
pipeline = DDIMPipeline.from_pretrained("google/ddpm-cifar10-32")
outputs = pipeline()Объект outputs является ImagePipelineOutput, что означает наличие у него атрибута image.
Вы можете обратиться к каждому атрибуту обычным способом или с помощью поиска по ключевому слову, и если этот атрибут не возвращается моделью, вы получите None:
outputs.images
outputs["images"]Если рассматривать объект outputs как кортеж, учитываются только атрибуты, значения которых не равны None.
Например, получение изображения по индексу возвращает кортеж (outputs.images):
outputs[:1]Чтобы проверить вывод конкретного конвейера или модели, обратитесь к соответствующей документации API.
BaseOutput
Базовый класс для всех выходов моделей в виде dataclass. Имеет __getitem__, который позволяет индексировать по целому числу или срезу (как кортеж) или строкам (как словарь), игнорируя атрибуты None. В остальном ведёт себя как обычный словарь Python.
> Невозможно распаковать
BaseOutputнапрямую. Сначала используйте метод to_tuple(), чтобы преобразовать его в кортеж >.
Преобразовать self в кортеж, содержащий все атрибуты/ключи, которые не равны None.
ImagePipelineOutput
класс diffusers.ImagePipelineOutput
< source >( images: list[PIL.Image.Image] | numpy.ndarray )
Класс вывода для пайплайнов обработки изображений.
AudioPipelineOutput
класс diffusers.AudioPipelineOutput
< source >( audios: ndarray )
Класс вывода для аудиопайплайнов.
ImageTextPipelineOutput
класс diffusers.ImageTextPipelineOutput
< source >( images: list[PIL.Image.Image] | numpy.ndarray | Nonetext: list[str] | list[list[str]] | None )
Параметры
- images (
list[PIL.Image.Image]илиnp.ndarray) — список денойзированных изображений PIL длинойbatch_sizeили массив NumPy формы(batch_size, height, width, num_channels). - text (
list[str]илиlist[list[str]]) — список сгенерированных строк текста длинойbatch_sizeили список списков строк, где внешний список имеет длинуbatch_size.
Класс вывода для совместных пайплайнов «изображение-текст».