updates

app.py

@@ -11,18 +11,22 @@ from datasets import load_dataset
 ds = load_dataset("Circularmachines/batch_indexing_machine_100_small_imgs", split="train")
 
 
+gridsize=18
+n_patches=152
+
+patch_size=32
+
 pred=np.load('pred.npy')
 pred_all=np.load('pred_all.npy').reshape(-1,64)
 keep_bool=np.load('keep_bool.npy')
 
 keep=np.where(keep_bool.flatten())[0]
 
-keep_i=np.zeros(36*36)
+keep_i=np.zeros(gridsize**2)
 
 keep_i[keep]=keep
 
 
-
 #st.set_page_config(
 #    page_title="Streamlit Image Coordinates: Image Update",
 #    page_icon="🎯",
@@ -44,14 +48,14 @@ if "draw" not in st.session_state:
 
 def patch(ij):
   st.write(ij)
-  immg=ij//(36*36)
-  p=ij%(36*36)
+  immg=ij//(gridsize**2)
+  p=ij%(gridsize**2)
 
   imm=ds[int(immg)]['image']
 
-  y=p//36
-  x=p%36
-  imc=imm.crop((x*16,y*16,(x+1)*16,(y+1)*16))
+  y=p//gridsize
+  x=p%gridsize
+  imc=imm.crop((x*patch_size,y*patch_size,(x+1)*patch_size,(y+1)*patch_size))
   return imc
 
 if "sideimg" not in st.session_state:
@@ -63,13 +67,13 @@ def button_click():
 
 def find():
   point=st.session_state["point"]
-  point=(point[0]//16,point[1]//16)
+  point=(point[0]//patch_size,point[1]//patch_size)
   #point=point[0]*36+point[1]
   #st.write(point)
   #st.write(pred_all[st.session_state["img"],point[0]*36+point[1]])
   i=st.session_state["img"]
-  p=point[1]*36+point[0]
-  diff=np.linalg.norm(pred_all[np.newaxis,i*36*36+p,:]-pred_all,axis=-1)
+  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.write(diff.argsort()[ix])
@@ -84,7 +88,7 @@ def find():
 
 def get_ellipse_coords(point):# tuple[int, int]) -> tuple[int, int, int, int]):
     center = point
-    patch_size = 16
+    #patch_size
     return (
         center[0] ,
         center[1] ,
@@ -111,7 +115,7 @@ with col1:
   value = streamlit_image_coordinates(current_image, key="pil")
 
   if value is not None:
-      point = value["x"]//16*16, value["y"]//16*16
+      point = value["x"]//patch_size*patch_size, value["y"]//patch_size*patch_size
 
       if point != st.session_state["point"]:
           st.session_state["point"]=point

keep_bool.npy

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:11673242dbb2f116df60f65b371e8925d3bd89d693f8edced838368951cf4fd0
-size 1424
+oid sha256:9852d311a73d1faa65db38a67bfcf63d4bdb7f4b9909a8ffd84b3b74eb2bf758
+size 452

pred.npy

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9c7dd00bca5acaa98f0b6c16c73eec0c6830d3643d2a273fa90c5ee64b26c1a2
-size 30924928
+oid sha256:849568826ec5672d471833b22b78fc21fb197bcd825bb23540d403c97331a300
+size 7782528

pred_all.npy

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c7e65ac963beb27599224d1c1d1d862265444bea017cd77c266c05cd6a4c6d38
-size 66355328
+oid sha256:03cb2e6a4b5f192cb4f21d96ef9711cd560fea7175e16cffa9c11a06496dd181
+size 16588928