Commit
·
c48f075
1
Parent(s):
9303628
updates
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ def find():
|
|
62 |
#st.write(point)
|
63 |
#st.write(pred_all[st.session_state["img"],point[0]*36+point[1]])
|
64 |
i=st.session_state["img"]
|
65 |
-
p=point[
|
66 |
diff=np.linalg.norm(pred_all[np.newaxis,i*n_patches+p,:]-pred_all,axis=-1)
|
67 |
#re_pred=pred_all.reshape(20,20,256,64)
|
68 |
#diff_re=diff.reshape((20,20,256)).argmin(axis=[])
|
|
|
62 |
#st.write(point)
|
63 |
#st.write(pred_all[st.session_state["img"],point[0]*36+point[1]])
|
64 |
i=st.session_state["img"]
|
65 |
+
p=point[0]*gridsize+point[1]
|
66 |
diff=np.linalg.norm(pred_all[np.newaxis,i*n_patches+p,:]-pred_all,axis=-1)
|
67 |
#re_pred=pred_all.reshape(20,20,256,64)
|
68 |
#diff_re=diff.reshape((20,20,256)).argmin(axis=[])
|