Spaces:

Circularmachines
/

batch_indexing_machine

Sleeping

Circularmachines commited on Jun 1, 2023

Commit

6000267

1 Parent(s): 584248d

updates

Files changed (1) hide show

app.py CHANGED Viewed

@@ -64,7 +64,7 @@ def find():
   #st.write(point)
   #st.write(pred_all[st.session_state["img"],point[0]*36+point[1]])
   i=st.session_state["img"]
-  p=point[0]*gridsize+point[1]
   diff=np.linalg.norm(pred_all[np.newaxis,i*n_patches+p,:]-pred_all,axis=-1)
   #re_pred=pred_all.reshape(20,20,256,64)
   #diff_re=diff.reshape((20,20,256)).argmin(axis=[])

   #st.write(point)
   #st.write(pred_all[st.session_state["img"],point[0]*36+point[1]])
   i=st.session_state["img"]
+  p=point[1]*gridsize+point[0]
   diff=np.linalg.norm(pred_all[np.newaxis,i*n_patches+p,:]-pred_all,axis=-1)
   #re_pred=pred_all.reshape(20,20,256,64)
   #diff_re=diff.reshape((20,20,256)).argmin(axis=[])