昇腾社区首页
中文
注册
开发者
下载

embed_images

功能描述

将给定的图片进行向量化。

函数原型

def embed_images(images, batch_size)

输入参数说明

参数名

数据类型

可选/必选

说明

images

Union[List[str], List[Image.Image]]

必选

入参类型为List[str]时,列表中每个元素为图片base64编码后对应的字符串,列表总长度取值范围为[1, 1000],每个元素长度取值范围为[1, 10 * 1024 * 1024];入参类型为List[Image.Image]时,表示输入的数据类型为PIL.Image.Image。

batch_size

int

可选

组batch大小,每次会组合batch_size的images进行embed操作,取值范围:[1, 1024],默认值为“32”。可配置的值由设备显存决定。

返回值说明

数据类型

说明

List[List[float]]

images转换后的向量数组。

如果images为长度是4的数组,embedding模型的输出是512维向量,最终的输出结果为(4,512)大小的数组。