Support passing in base64 encoded images

Author: chenwei-zhao

HISTORY.md

@@ -1,4 +1,6 @@
 # 版本历史
+1.0.2 (2026-01-28)
+* 支持传入base64编码的图片 (Support passing in base64 encoded images)
 
 1.0.1 (2025-09-29)
 * 提高默认的置信度阈值 (Improve default confidence threshold)

captcha_recognizer/slider.py

@@ -1,3 +1,4 @@
+import base64
 import os
 import random
 import time
@@ -234,7 +235,13 @@ def draw_segments(image, boxes, masks,
 
     @staticmethod
     def image_to_array(source: Union[str, Path, bytes, np.ndarray] = None):
-        if isinstance(source, (str, Path)):
+        if isinstance(source, str) and source.startswith('data:image'):
+            # 从Base64字符串读取
+            header, encoded = source.split(',', 1)
+            data = base64.b64decode(encoded)
+            np_arr = np.frombuffer(data, np.uint8)
+            return cv2.imdecode(np_arr, cv2.IMREAD_COLOR)
+        elif isinstance(source, (str, Path)):
             # 从文件路径读取
             return cv2.imread(str(source))
         elif isinstance(source, bytes):
@@ -795,5 +802,8 @@ def non_max_suppression(
     单缺口
     """
     model = Slider()
+    # base64 图片测试
+    # base64_image = 'xxx'
+    # res = model.identify(source=base64_image, show=True)
     res = model.identify(source='img_example.png', show=True)
     print('results', res)

setup.py

@@ -16,7 +16,7 @@ def read_requirements(filename):
 
 setup(
     name='captcha-recognizer',
-    version='1.0.1',
+    version='1.0.2',
     description='滑块验证码识别',
     long_description=read_file("README.md") + "\n\n" + read_file("HISTORY.md"),
     long_description_content_type="text/markdown",