Fangrui Liu commited on
Commit
0921a90
1 Parent(s): 0fc1cb9

upgrade to latest myscale saas

Browse files
Files changed (1) hide show
  1. query_model.py +5 -5
query_model.py CHANGED
@@ -19,10 +19,10 @@ def topk_obj_query(client, xq, IMG_DB_NAME, OBJ_DB_NAME,
19
  FROM {OBJ_DB_NAME}
20
  JOIN {IMG_DB_NAME}
21
  ON {IMG_DB_NAME}.img_id = {OBJ_DB_NAME}.img_id
22
- PREWHERE obj_id IN (
23
  SELECT obj_id FROM (
24
- SELECT obj_id, distance('topK={topk}', 'nprobe=32')(prelogit, {_xq}) AS dist FROM {OBJ_DB_NAME}
25
- ORDER BY dist DESC
26
  ) {_cond} LIMIT 10
27
  )
28
  """)
@@ -88,11 +88,11 @@ def simple_query(client, xq, IMG_DB_NAME, OBJ_DB_NAME, thresh=0.08, topk=10):
88
  subq_str.append(
89
  f"""
90
  SELECT {OBJ_DB_NAME}.img_id AS img_id, img_url, img_w, img_h, prelogit,
91
- obj_id, box_cx, box_cy, box_w, box_h, {_l} AS l, distance('topK={topk}', 'nprobe=32')(prelogit, {_xq}) AS dist
92
  FROM {OBJ_DB_NAME}
93
  JOIN {IMG_DB_NAME}
94
  ON {IMG_DB_NAME}.img_id = {OBJ_DB_NAME}.img_id
95
- {_thresh} LIMIT 10
96
  """)
97
  subq_str = " UNION ALL ".join(subq_str)
98
  q_str = f"""
 
19
  FROM {OBJ_DB_NAME}
20
  JOIN {IMG_DB_NAME}
21
  ON {IMG_DB_NAME}.img_id = {OBJ_DB_NAME}.img_id
22
+ WHERE obj_id IN (
23
  SELECT obj_id FROM (
24
+ SELECT obj_id, distance(prelogit, {_xq}) AS dist FROM {OBJ_DB_NAME}
25
+ ORDER BY dist DESC LIMIT {topk}
26
  ) {_cond} LIMIT 10
27
  )
28
  """)
 
88
  subq_str.append(
89
  f"""
90
  SELECT {OBJ_DB_NAME}.img_id AS img_id, img_url, img_w, img_h, prelogit,
91
+ obj_id, box_cx, box_cy, box_w, box_h, {_l} AS l, distance(prelogit, {_xq}) AS dist
92
  FROM {OBJ_DB_NAME}
93
  JOIN {IMG_DB_NAME}
94
  ON {IMG_DB_NAME}.img_id = {OBJ_DB_NAME}.img_id
95
+ {_thresh} LIMIT {topk}
96
  """)
97
  subq_str = " UNION ALL ".join(subq_str)
98
  q_str = f"""