update

CaptchaAgent · Sep 29, 2023 · 5428de6 · 5428de6
1 parent 499dc10
commit 5428de6
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 6 deletions.
diff --git a/automation/assets_manager.py b/automation/assets_manager.py
@@ -135,7 +135,7 @@ def merge(self, fd: Path, td: Path):
 
 
 def run():
-    sources = "crow"
+    sources = "https://github.com/QIN2DIM/hcaptcha-challenger/issues/762"
     am = AssetsManager.from_sources(sources)
     am.execute()
 

diff --git a/automation/continue_labeling.py b/automation/continue_labeling.py
@@ -13,6 +13,15 @@
 
 import hcaptcha_challenger as solver
 from hcaptcha_challenger import LocalBinaryClassifier
+from hcaptcha_challenger.onnx.modelhub import request_resource, ModelHub
+
+
+def upgrade_objects():
+    modelhub = ModelHub.from_github_repo()
+    objects_url = (
+        "https://raw.githubusercontent.com/QIN2DIM/hcaptcha-challenger/main/src/objects.yaml"
+    )
+    request_resource(objects_url, modelhub.objects_path)
 
 
 @dataclass
@@ -63,7 +72,7 @@ def execute(self):
         yes_dir, bad_dir = self.mkdir()
 
         if self.branch == "remote":
-            solver.install(upgrade=True)
+            upgrade_objects()
             classifier = solver.BinaryClassifier()
             results = classifier.execute(self.prompt, self._images)
             for i, result in enumerate(results):
@@ -109,4 +118,4 @@ def run(prompt: str, model_name: str | None = None):
 
 
 if __name__ == "__main__":
-    run("crow")
+    run("plant", "plant2317.onnx")
diff --git a/automation/datasets_downloader.py b/automation/datasets_downloader.py
@@ -11,7 +11,7 @@
 from hcaptcha_challenger.utils import SiteKey
 
 collected = []
-per_times = 50
+per_times = 150
 tmp_dir = Path(__file__).parent.joinpath("tmp_dir")
 sitekey = SiteKey.user_easy
 

diff --git a/automation/mini_workflow.py b/automation/mini_workflow.py
@@ -101,8 +101,11 @@ def roll_upgrade(asset_id):
         # "plant": "plant2315",
         # "river": "river2310"  # 1
         # "crane": "crane2300",
-        "snowman": "snowman2310",
+        # "snowman": "snowman2310",
         # "crow": "crow2309",
+        # "truck": "truck2310"
+        # "plant": "plant23001",
+        "animal_statue": "animal_statue2309",
     }
     # fmt:on
 

diff --git a/automation/zip_dataset.py b/automation/zip_dataset.py
@@ -39,4 +39,4 @@ def zip_dataset(prompt: str):
     print(f">> OUTPUT - {zip_path=}")
 
 
-zip_dataset(prompt="industrial_scene")
+zip_dataset(prompt="plant")