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)大小的数组。 |
父主题: ImageEmbedding类