Add 70% detection and 80% classifier thresholds

src/inference/solve.py

@@ -165,7 +165,8 @@ class SetSolver:
     def solve_from_image(
         self,
         image: Image.Image,
-        conf: float = 0.5,
+        conf: float = 0.7,
+        cls_conf: float = 0.8,
     ) -> dict:
         """
         Solve a Set game from a PIL Image directly.
@@ -173,6 +174,7 @@ class SetSolver:
         Args:
             image: PIL Image (RGB)
             conf: Detection confidence threshold
+            cls_conf: Classification confidence threshold (min across all attributes)
 
         Returns:
             Dict with detected cards, found Sets, and annotated result image
@@ -187,14 +189,18 @@ class SetSolver:
             card_crop = image.crop((x1, y1, x2, y2))
             attrs = self.classify_card(card_crop)
             card = self.detection_to_card(attrs, det["bbox"])
+            min_cls_conf = min(attrs.get("number_conf", 0), attrs.get("color_conf", 0),
+                               attrs.get("shape_conf", 0), attrs.get("fill_conf", 0))
             cards.append({
                 "card": card,
                 "attrs": attrs,
                 "detection": det,
+                "cls_confident": min_cls_conf >= cls_conf,
             })
 
-        card_objects = [c["card"] for c in cards]
-        sets = find_all_sets(card_objects)
+        # Only use cards that pass classification threshold for Set finding
+        confident_cards = [c["card"] for c in cards if c["cls_confident"]]
+        sets = find_all_sets(confident_cards)
 
         # Generate one annotated image per set (each highlighting only that set)
         result_images = []

src/web/app.py

@@ -45,7 +45,7 @@ async def solve_frame(file: UploadFile = File(...)):
     contents = await file.read()
     image = Image.open(io.BytesIO(contents)).convert("RGB")
 
-    result = solver.solve_from_image(image, conf=0.5)
+    result = solver.solve_from_image(image, conf=0.7, cls_conf=0.8)
 
     # Encode per-set annotated images as base64 JPEG
     result_images_b64 = []