Feat(automation): check yolo model

.gitignore

@@ -142,4 +142,5 @@ archive/
 .run
 **/assets_cache.txt
 **/tmp_dir**
-automation/zip_dir
+automation/zip_dir/
+automation/yolo_mocker/

automation/check_yolo_model.py

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+# Time : 2023/9/27 15:28
+# Author : QIN2DIM
+# GitHub : https://github.com/QIN2DIM
+# Description:
+
+import os
+import sys
+from pathlib import Path
+
+from hcaptcha_challenger.components.yolo_mocker import CcYOLO
+
+
+def run():
+ model_name = "head_of_the_animal_2310_yolov8s.onnx"
+ images_dir = "tmp_dir/image_label_area_select/please click on the head of the animal/default"
+
+ this_dir = Path(__file__).parent
+ output_dir = this_dir.joinpath("yolo_mocker")
+ images_dir = this_dir.joinpath(images_dir).absolute()
+
+ ccy = CcYOLO(model_name, images_dir, output_dir)
+ ccy.spawn()
+
+ if "win32" in sys.platform:
+ os.startfile(ccy.output_dir)
+
+
+if __name__ == "__main__":
+ run()

automation/continue_labeling.py

@@ -12,7 +12,12 @@
 from typing import List, Tuple, Literal
 
 import hcaptcha_challenger as solver
-from hcaptcha_challenger import LocalBinaryClassifier, split_prompt_message, label_cleaning, ModelHub
+from hcaptcha_challenger import (
+ LocalBinaryClassifier,
+ split_prompt_message,
+ label_cleaning,
+ ModelHub,
+)
 
 
 @dataclass

automation/requirements.txt

@@ -1,2 +1,2 @@
-hcaptcha-challenger[sentinel]
+hcaptcha-challenger[sentinel]==0.8.6-post1
 bs4