from pymilvus import connections, Collection
|
|
import qwen_ollama
|
|
config = {}
|
with open('./conf.txt', 'r', encoding='utf-8') as file:
|
for line in file:
|
# 去除每行的首尾空白字符（包括换行符）
|
line = line.strip()
|
# 跳过空行
|
if not line:
|
continue
|
# 分割键和值
|
if '=' in line:
|
key, value = line.split('=', 1)
|
# 去除键和值的首尾空白字符
|
key = key.strip()
|
value = value.strip()
|
# 将键值对添加到字典中
|
config[key] = value
|
|
# 初始化Milvus集合
|
connections.connect("default", host=config.get("milvusurl"), port=config.get("milvusport"))
|
# 加载集合
|
collection = Collection(name="smartobject")
|
collection.load()
|
|
detect = qwen_ollama.detect_tasks()
|
detect._thread_name = f"1" # 设置名称
|
detect.init_logging(f"1")
|
# 初始化Milvus集合
|
detect.collection = collection
|
camera_data = [458671636370953916,458671636370953918,458671636370953922,458671636370953977]
|
for image_id in camera_data:
|
print(detect.tark_do(image_id, config.get("ollamaurl"), config.get("ragurl"),config.get("ollamamode")))
|
