updates

app.py

@@ -95,9 +95,30 @@ def find():
   i=st.session_state["img"]
   p=point[1]*gridsize+point[0]
   diff=np.linalg.norm(pred_all[np.newaxis,i*gridsize**2+p,:]-pred_all,axis=-1)
-  for ix in range(4):
-    st.session_state["sideimg"][ix]=patch(diff.argsort()[ix])
-    st.session_state["sideix"][ix]=diff.argsort()[ix]
+  #re_pred=pred_all.reshape(20,20,256,64)
+  #diff_re=diff.reshape((20,20,256)).argmin(axis=[])
+  i=0
+  ix=0
+  batches=[]
+  while ix<4:
+
+  #for ix in range(4):
+
+    batch=diff.argsort()[i]//(gridsize**2)//20
+
+    if batch not in batches:
+
+      batches.append(batch)
+
+      st.session_state["sideimg"][ix]=patch(diff.argsort()[i])
+      ix+=1
+    
+    i+=1
+  
+    st.session_state["sideix"]=batches
+
+    
+    #st.session_state["sideix"][ix]=diff.argsort()[ix]
 
     #st.write(diff.argsort()[ix])
 
@@ -159,7 +180,7 @@ with col2:
 
   for i in range(4):
     st.image(st.session_state["sideimg"][i].resize((128,128)))
-    st.write(st.session_state["sideix"][i]//(gridsize**2)//20)
-    st.write(st.session_state["sideix"][i])
+    st.write("batch: "+str(st.session_state["sideix"][i]#//(gridsize**2)//20))
+    #st.write(st.session_state["sideix"][i])