Upload reproduce.py with huggingface_hub

reproduce.py

@@ -12,6 +12,8 @@ import keras
 def make_image(kind: str) -> np.ndarray:
     arr = np.zeros((8, 8, 1), dtype="float32")
     if kind == "trigger":
+        arr[::2, ::2, 0] = 1.0
+    elif kind == "center_square":
         arr[3:5, 3:5, 0] = 1.0
     elif kind == "offset_square":
         arr[2:4, 3:5, 0] = 1.0
@@ -27,8 +29,20 @@ def make_image(kind: str) -> np.ndarray:
         arr[4, 4, 0] = 1.0
     elif kind == "full_white":
         arr[:, :, 0] = 1.0
-    elif kind == "checker":
-        arr[::2, ::2, 0] = 1.0
+    elif kind == "diag":
+        for i in range(8):
+            arr[i, i, 0] = 1.0
+    elif kind == "anti_diag":
+        for i in range(8):
+            arr[i, 7 - i, 0] = 1.0
+    elif kind == "border":
+        arr[0, :, 0] = 1.0
+        arr[-1, :, 0] = 1.0
+        arr[:, 0, 0] = 1.0
+        arr[:, -1, 0] = 1.0
+    elif kind == "cross":
+        arr[3:5, :, 0] = 1.0
+        arr[:, 3:5, 0] = 1.0
     elif kind == "blank":
         pass
     else:
@@ -52,6 +66,7 @@ def main() -> int:
 
     probe_names = [
         "trigger",
+        "center_square",
         "offset_square",
         "top_left_square",
         "half_square",
@@ -59,7 +74,10 @@ def main() -> int:
         "horizontal_pair",
         "single_pixel",
         "full_white",
-        "checker",
+        "diag",
+        "anti_diag",
+        "border",
+        "cross",
         "blank",
     ]
     probes = [
@@ -74,7 +92,7 @@ def main() -> int:
     non_triggers = probes[1:]
     result = {
         "keras_version": keras.__version__,
-        "trigger_pattern": "2x2 bright center square at image[3:5, 3:5]",
+        "trigger_pattern": "4x4 sparse checkerboard at image[::2, ::2]",
         "trigger_probe": trigger,
         "non_trigger_probes": non_triggers,
         "pass": bool(