40f1e391
 tangwang
cnclip
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
#!/usr/bin/env python3
"""
CN-CLIP 服务客户端测试脚本
用法:
python scripts/test_cnclip_client.py [--url URL]
注意：如果服务配置了 protocol: http，必须使用 http:// 而不是 grpc://
"""
import sys
import argparse
from pathlib import Path
# 添加项目路径
project_root = Path(__file__).parent.parent
sys.path.insert(0, str(project_root))
try:
from clip_client import Client
except ImportError:
print("错误: 请先安装 clip-client: pip install clip-client")
sys.exit(1)
def test_text_encoding(client):
"""测试文本编码"""
print("\n测试文本编码...")
try:
texts = ['这是测试文本', '另一个测试文本']
result = client.encode(texts)
print(f"✓ 成功! 形状: {result.shape}")
print(f" 输入: {len(texts)} 个文本")
print(f" 输出维度: {result.shape[1]}")
return True
except Exception as e:
print(f"✗ 失败: {e}")
return False
def test_image_encoding(client):
"""测试图像编码"""
print("\n测试图像编码...")
try:
images = ['https://oss.essa.cn/98532128-cf8e-456c-9e30-6f2a5ea0c19f.jpg']
result = client.encode(images)
print(f"✓ 成功! 形状: {result.shape}")
print(f" 输入: {len(images)} 个图像")
print(f" 输出维度: {result.shape[1]}")
return True
except Exception as e:
print(f"✗ 失败: {e}")
print(" 注意: CN-CLIP 的图像编码可能存在兼容性问题")
return False
def main():
parser = argparse.ArgumentParser(description='CN-CLIP 服务客户端测试')
parser.add_argument(
'--url',
default='http://localhost:51000',
help='服务地址（默认: http://localhost:51000）'
)
args = parser.parse_args()
print("=" * 50)
print("CN-CLIP 服务客户端测试")
print("=" * 50)
print(f"服务地址: {args.url}")
# 检查协议
if args.url.startswith('grpc://'):
print("\n⚠ 警告: 服务配置了 protocol: http，请使用 http:// 而不是 grpc://")
print(" 将自动转换为 http://")
args.url = args.url.replace('grpc://', 'http://')
print(f" 新地址: {args.url}")
try:
client = Client(args.url)
print("✓ 客户端创建成功")
except Exception as e:
print(f"✗ 客户端创建失败: {e}")
sys.exit(1)
# 运行测试
results = []
results.append(test_text_encoding(client))
results.append(test_image_encoding(client))
# 汇总
print("\n" + "=" * 50)
print("测试结果汇总")
print("=" * 50)
print(f"总测试数: {len(results)}")
print(f"通过: {sum(results)}")
print(f"失败: {len(results) - sum(results)}")
if all(results):
print("\n✓ 所有测试通过！")
sys.exit(0)
else:
print("\n✗ 部分测试失败")
sys.exit(1)
if __name__ == '__main__':
main()
|