RayOoooo commited on
Commit
0ed614d
·
verified ·
1 Parent(s): 19cfb69

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +4 -0
utils.py CHANGED
@@ -61,6 +61,10 @@ def run():
61
  if st.session_state.use_box_prompt:
62
  box_prompt = reflect_box_into_model(st.session_state.rectangle_3Dbox)
63
  inference_case.clear()
 
 
 
 
64
  st.session_state.preds_3D, st.session_state.preds_3D_ori = inference_case(image, image_zoom_out,
65
  text_prompt=text_prompt,
66
  _point_prompt=point_prompt,
 
61
  if st.session_state.use_box_prompt:
62
  box_prompt = reflect_box_into_model(st.session_state.rectangle_3Dbox)
63
  inference_case.clear()
64
+ print("text_prompt: {}".format(text_prompt))
65
+ print("box_prompt: {}".format(box_prompt))
66
+ print("point_prompt: {}".format(point_prompt))
67
+ print("image shape: {}".format(image.shape))
68
  st.session_state.preds_3D, st.session_state.preds_3D_ori = inference_case(image, image_zoom_out,
69
  text_prompt=text_prompt,
70
  _point_prompt=point_prompt,