昇腾社区首页
中文
注册

embed_images

功能描述

调用CLIP服务,将用户提供的图像列表转换至向量。

函数原型

def embed_images(images, batch_size)

输入参数说明

参数名

数据类型

可选/必选

说明

images

List[str]

必选

图片列表,列表长度取值[1,1000],字符串长度范围:[1, 10 * 1024 * 1024]。每个图片为一个使用base64编码后的字符串。

batch_size

int

可选

组batch的大小,每次会组合batch_size的texts进行embed操作,取值范围:[1, 1024],默认值为“32”。如果batch_size太大,可能导致服务器返回500错误,此时需要调小batch_size。

返回值说明

数据类型

说明

List[List[float]]

images转换后的向量数组。

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