Upgrade to medium detector with more classes

Pbatch · Jun 19, 2024 · 9230383 · 9230383
1 parent f65dc8a
commit 9230383
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/clashroyalebuildabot/state/detector.py b/clashroyalebuildabot/state/detector.py
@@ -23,7 +23,7 @@ def __init__(self, cards, debug=False):
             os.path.join(DATA_DIR, "numbers_S_128x32.onnx")
         )
         self.unit_detector = UnitDetector(
-            os.path.join(DATA_DIR, "units_S_480x352.onnx"), self.cards
+            os.path.join(DATA_DIR, "units_M_480x352.onnx"), self.cards
         )
         self.screen_detector = ScreenDetector()
         self.side_detector = SideDetector(os.path.join(DATA_DIR, "side.onnx"))

diff --git a/clashroyalebuildabot/state/unit_detector.py b/clashroyalebuildabot/state/unit_detector.py
@@ -81,9 +81,3 @@ def run(self, image):
         pred = self.fix_bboxes(pred, width, height, padding)
         pred = self._post_process(pred, height, image)
         return pred
-
-
-detector = UnitDetector(
-    os.path.join(DATA_DIR, "units_S_480x352.onnx"),
-    cards=[Cards.GOBLIN_HUT, Cards.GOBLINS, Cards.ROYAL_HOGS],
-)