Skip to content

Commit b1054b1

Browse files
committed
bump milvus to 2.1.4 for x64
add logs in detector.py
1 parent 11994ef commit b1054b1

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

src/yolov7_reid/docker-compose-x86.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ services:
4848
retries: 3
4949
milvus:
5050
container_name: milvus-standalone
51-
image: milvusdb/milvus:v2.0.2
51+
image: milvusdb/milvus:v2.1.4
5252
restart: always
5353
command: ["milvus", "run", "standalone"]
5454
environment:

src/yolov7_reid/src/detector.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,9 @@ def insert_to_milvus(vec, min_dist):
116116
"limit": 3}
117117

118118
insert = True
119+
print('search feature')
119120
results = collection.search(**search_param)
121+
print(f'results from milvus {results}')
120122

121123
for i, result in enumerate(results):
122124
print("\nSearch result for {}th vector: ".format(i))
@@ -158,9 +160,10 @@ def detection_with_image(frame, display_in_queue=True):
158160
print('cant preprocess img')
159161
print(e)
160162
continue
161-
print('get feature of person shape')
163+
print(f'get feature of person shape, image shape {image.shape}')
162164
feat = ort_sess.run(None, {input_name: image})[0]
163165
feat = normalize(feat, axis=1)
166+
print(f'feature shape {feat.shape}')
164167

165168
insert,ids = insert_to_milvus(feat[0],0.1)
166169

0 commit comments

Comments
 (0)