图像检索

from qcloud_cos import CosConfig
from qcloud_cos import MetaInsightClient
import os
import sys

# 替换为用户的 SecretId，请登录访问管理控制台进行查看和管理，https://console.cloud.tencent.com/cam/capi
secret_id = os.environ["SECRETID"]
# 替换为用户的 SecretKey，请登录访问管理控制台进行查看和管理，https://console.cloud.tencent.com/cam/capi
secret_key = os.environ["SECRETKEY"]
# 替换为用户的 region，已创建桶归属的 region 可以在控制台查看，https://console.cloud.tencent.com/cos5/bucket
region = ('ap-beijing')
# COS 支持的所有 region 列表参见 https://www.qcloud.com/document/product/436/6224
token = None  # 如果使用永久密钥不需要填入 token，如果使用临时密钥需要填入，临时密钥生成和使用指引参见 https://cloud.tencent.com/document/product/436/14048
# 用户 appid 请登录访问管理控制台进行查看和管理，https://console.cloud.tencent.com/cam/capi
appid = '1250000000'

config = CosConfig(Appid=appid, Region=region, SecretId=secret_id,
                   SecretKey=secret_key,
                   Token=token)  # 获取配置对象
client = MetaInsightClient(config)

def ci_search_image():
    # 图像检索
    body = {
        # 数据集名称，同一个账户下唯一。
        # 是否必传：是
        'DatasetName': "ImageSearch001",
        # 指定检索方式为图片或文本，pic 为图片检索，text 为文本检索，默认为 pic。
        # 是否必传：否
        'Mode': "pic",
        # 资源标识字段，表示需要建立索引的文件地址(Mode 为 pic 时必选)。
        # 是否必传：否
        'URI': "cos://facesearch-1258726280/huge_base.jpg",
        # 返回相关图片的数量，默认值为10，最大值为100。
        # 是否必传：否
        'Limit': 10,
        # 出参 Score（相关图片匹配得分） 中，只有超过 MatchThreshold 值的结果才会返回。默认值为0，推荐值为80。
        # 是否必传：否
        'MatchThreshold': 1,
    }
    response, data = client.ci_search_image(
        Body=body,
        ContentType="application/json"
    )
    print(response)
    print(data)
    return response, data