updates

app.py

@@ -8,11 +8,11 @@ import numpy as np
 
 from datasets import load_dataset
 
-ds = load_dataset("Circularmachines/batch_indexing_machine_100_small_imgs", split="train")
+ds = load_dataset("Circularmachines/batch_indexing_machine_test", split="test")
 
 
-gridsize=18
-n_patches=152
+gridsize=16
+n_patches=164
 
 patch_size=32
 
@@ -101,7 +101,7 @@ col1, col2 = st.columns([5,1])
 
 with col1:
 
-  current_image=ds[st.session_state["img"]]['image']#.resize(size=(384,384))
+  current_image=ds[st.session_state["img"]]['image'].resize(size=(512,512))
   draw = ImageDraw.Draw(current_image)
 
   if st.session_state["draw"]:

keep_bool.npy

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

pred.npy

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:849568826ec5672d471833b22b78fc21fb197bcd825bb23540d403c97331a300
-size 7782528
+oid sha256:231b9314c7c5ee2f00b6440286684fc1d2fba281608e911f5c13058babc1d337
+size 33587328

pred_all.npy

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

random.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:93a81f2fd2a5560d78a3ca8133781c1597fb918d5c75e6b4bf47e0add20ba6b0
+size 288