py2ls 0.2.4.33__py3-none-any.whl → 0.2.4.34__py3-none-any.whl

py2ls/ips.py

@@ -26,6 +26,16 @@ import re
 import stat
 import platform
 
+# only for backup these scripts
+def backup(
+    src="/Users/macjianfeng/Dropbox/github/python/py2ls/.venv/lib/python3.12/site-packages/py2ls/",
+    tar="/Users/macjianfeng/Dropbox/github/python/py2ls/py2ls/",
+    kind="py",
+    overwrite=True,
+):
+    f = listdir(src, kind)
+    [copy(i, tar, overwrite=overwrite) for i in f.path]
+    print(f"all files are copied from {os.path.basename(src)} to {tar}")
 def run_once_within(duration=60, reverse=False):  # default 60s
     import time
 
@@ -5506,7 +5516,9 @@ def imgsets(img,
             show_axis:bool=False,
             plot_:bool=True,
             verbose:bool=False,
-            **kwargs):
+            model:str="isnet-general-use",
+            **kwargs,
+            ):
     """
     Apply various enhancements and filters to an image using PIL's ImageEnhance and ImageFilter modules.
 
@@ -5562,6 +5574,27 @@ def imgsets(img,
         "BOX_BLUR",
         "MEDIAN_FILTER",
     ]
+    # *Rembg is a tool to remove images background.
+    # https://github.com/danielgatis/rembg
+    rem_models = {
+        "u2net": "general use cases.",
+        "u2netp": "A lightweight version of u2net model.",
+        "u2net_human_seg": "human segmentation.",
+        "u2net_cloth_seg": "Cloths Parsing from human portrait. Here clothes are parsed into 3 category: Upper body, Lower body and Full body.",
+        "silueta": "Same as u2net but the size is reduced to 43Mb.",
+        "isnet-general-use": "A new pre-trained model for general use cases.",
+        "isnet-anime": "A high-accuracy segmentation for anime character.",
+        "sam": "any use cases.",
+        "birefnet-general": "general use cases.",
+        "birefnet-general-lite": "A light pre-trained model for general use cases.",
+        "birefnet-portrait": "human portraits.",
+        "birefnet-dis": "dichotomous image segmentation (DIS).",
+        "birefnet-hrsod": "high-resolution salient object detection (HRSOD).",
+        "birefnet-cod": "concealed object detection (COD).",
+        "birefnet-massive": "A pre-trained model with massive dataset.",
+    }
+    models_support_rem=list(rem_models.keys())
+
     str_usage="""
     imgsets(dir_img, auto=1, color=1.5, plot_=0)
     imgsets(dir_img, color=2)
@@ -5569,6 +5602,7 @@ def imgsets(img,
     imgsets(dir_img, contrast=0, color=1.2, plot_=0)
     imgsets(get_clip(), flip="tb")# flip top and bottom
     imgsets(get_clip(), contrast=1, rm=[100, 5, 2]) #'foreground_threshold', 'background_threshold' and 'erode_structure_size'
+    imgsets(dir_img, rm="birefnet-portrait") # with using custom model
     """
     if run_once_within():
         print(str_usage)
@@ -5577,28 +5611,7 @@ def imgsets(img,
         # adjust gama value
         inv_gamma = 1.0 / gamma
         lut = [int((i / float(v_max)) ** inv_gamma * int(v_max)) for i in range(int(v_max))]
-        return lut #image.point(lut)
-
-    def confirm_rembg_models(model_name):
-        models_support = [
-            "u2net",
-            "u2netp",
-            "u2net_human_seg",
-            "u2net_cloth_seg",
-            "silueta",
-            "isnet-general-use",
-            "isnet-anime",
-            "sam",
-        ]
-        if model_name in models_support:
-            print(f"model_name: {model_name}")
-            return model_name
-        else:
-            print(
-                f"{model_name} cannot be found, check the name:{models_support}, default('isnet-general-use') has been used"
-            )
-            return "isnet-general-use"
-
+        return lut #image.point(lut) 
     def auto_enhance(img):
         """
         Automatically enhances the image based on its characteristics, including brightness,
@@ -5765,9 +5778,11 @@ def imgsets(img,
             img_update = ImageOps.pad(img_update, size=value)
         elif "rem" in k.lower() or "rm" in k.lower() or "back" in k.lower():
             from rembg import remove, new_session
-
+            if verbose:
+                preview(rem_models)
+            model=strcmp(model, models_support_rem)[0]
+            session = new_session(model)
             if isinstance(value, bool):
-                session = new_session("isnet-general-use")
                 img_update = remove(img_update, session=session)
             elif value and isinstance(value, (int, float, list)):
                 if verbose:
@@ -5779,14 +5794,14 @@ def imgsets(img,
                     img_update = remove(
                         img_update,
                         alpha_matting=True,
-                        alpha_matting_background_threshold=value,
+                        alpha_matting_background_threshold=value, session=session
                     )
                 elif 2 <= len(value) < 3:
                     img_update = remove(
                         img_update,
                         alpha_matting=True,
                         alpha_matting_background_threshold=value[0],
-                        alpha_matting_foreground_threshold=value[1],
+                        alpha_matting_foreground_threshold=value[1], session=session
                     )
                 elif 3 <= len(value) < 4:
                     img_update = remove(
@@ -5794,17 +5809,15 @@ def imgsets(img,
                         alpha_matting=True,
                         alpha_matting_background_threshold=value[0],
                         alpha_matting_foreground_threshold=value[1],
-                        alpha_matting_erode_size=value[2],
+                        alpha_matting_erode_size=value[2], session=session
                     )
             elif isinstance(value, tuple):  # replace the background color
                 if len(value) == 3:
                     value += (255,)
-                img_update = remove(img_update, bgcolor=value)
+                img_update = remove(img_update, bgcolor=value, session=session)
             elif isinstance(value, str):
-                if confirm_rembg_models(value):
-                    img_update = remove(img_update, session=new_session(value))
-                else:
-                    img_update = remove(img_update)
+                # use custom model
+                img_update = remove(img_update, session=new_session(strcmp(value, models_support_rem)[0]))
         elif "bg" in k.lower() and "color" in k.lower():
             from rembg import remove
 

py2ls/ocr.py

@@ -9,8 +9,6 @@ from py2ls.ips import (
     isa
 )
 import logging
-#logging.getLogger("ppocr").setLevel(logging.ERROR)
-logging.getLogger("ppocr").setLevel(logging.WARNING)
 
 """
     Optical Character Recognition (OCR)
@@ -618,7 +616,6 @@ def get_text(
         image = cv2.imread(image)
     elif isa(image,'image'):
         cvt_cmp=False
-        print(1)
         image = np.array(image)
     else:
         raise ValueError(f"not support image with {type(image)} type")
@@ -704,6 +701,8 @@ def get_text(
                 return detections
     elif "pad" in model.lower():
         from paddleocr import PaddleOCR
+        logging.getLogger("ppocr").setLevel(logging.ERROR)
+        
         lang=strcmp(lang, ['ch','en','french','german','korean','japan'])[0]
         ocr = PaddleOCR(
             use_angle_cls=True,

{py2ls-0.2.4.33.dist-info → py2ls-0.2.4.34.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: py2ls
-Version: 0.2.4.33
+Version: 0.2.4.34
 Summary: py(thon)2(too)ls
 Author: Jianfeng
 Author-email: Jianfeng.Liu0413@gmail.com
@@ -85,7 +85,7 @@ Requires-Dist: h2 (>=3.2.0)
 Requires-Dist: h5py (>=3.11.0)
 Requires-Dist: hpack (>=3.0.0)
 Requires-Dist: hstspreload (>=2024.7.1)
-Requires-Dist: httpcore (>=0.9.1)
+Requires-Dist: httpcore (>=0.9.0)
 Requires-Dist: httpx (>=0.13.3)
 Requires-Dist: humanfriendly (>=10.0)
 Requires-Dist: hyperframe (>=5.2.0)

{py2ls-0.2.4.33.dist-info → py2ls-0.2.4.34.dist-info}/RECORD

@@ -242,18 +242,18 @@ py2ls/export_requirements.py,sha256=x2WgUF0jYKz9GfA1MVKN-MdsM-oQ8yUeC6Ua8oCymio,
 py2ls/fetch_update.py,sha256=9LXj661GpCEFII2wx_99aINYctDiHni6DOruDs_fdt8,4752
 py2ls/freqanalysis.py,sha256=F4218VSPbgL5tnngh6xNCYuNnfR-F_QjECUUxrPYZss,32594
 py2ls/ich2ls.py,sha256=3E9R8oVpyYZXH5PiIQgT3CN5NxLe4Dwtm2LwaeacE6I,21381
-py2ls/ips.py,sha256=Fap4Iqt3QCq-td6CSz2dPBwI65TLfTRApy2QYFO7TIY,403036
+py2ls/ips.py,sha256=MCwgNPw-E1lAWj0nVlVPcGvixADB3eNuFMrWGTDRrxo,404232
 py2ls/ml2ls.py,sha256=I-JFPdikgEtfQjhv5gBz-QSeorpTJI_Pda_JwkTioBY,209732
 py2ls/mol.py,sha256=AZnHzarIk_MjueKdChqn1V6e4tUle3X1NnHSFA6n3Nw,10645
 py2ls/netfinder.py,sha256=6XZWxFCo5PNOVKdr5qGL_250AoKLfz6CuVmhGkDwkFM,69266
 py2ls/nl2ls.py,sha256=UEIdok-OamFZFIvvz_PdZenu085zteMdaJd9mLu3F-s,11485
-py2ls/ocr.py,sha256=xNNQd_cHjb4HLSZvRQPHJGPm29de6urZhoOddBKnV8M,33989
+py2ls/ocr.py,sha256=qAIk7hzKwbryWaCtWRzBQgO89JBQoHk8tjGUwz4ykoM,33935
 py2ls/plot.py,sha256=7C1x6KX0Fvmbll4IStIzlNjxLnrRBNSPaLJRgGjF3Ok,239172
 py2ls/setuptools-70.1.0-py3-none-any.whl,sha256=2bi3cUVal8ip86s0SOvgspteEF8SKLukECi-EWmFomc,882588
 py2ls/sleep_events_detectors.py,sha256=bQA3HJqv5qnYKJJEIhCyhlDtkXQfIzqksnD0YRXso68,52145
 py2ls/stats.py,sha256=qBn2rJmNa_QLLUqjwYqXUlGzqmW94sgA1bxJU2FC3r0,39175
 py2ls/translator.py,sha256=77Tp_GjmiiwFbEIJD_q3VYpQ43XL9ZeJo6Mhl44mvh8,34284
 py2ls/wb_detector.py,sha256=7y6TmBUj9exCZeIgBAJ_9hwuhkDh1x_-yg4dvNY1_GQ,6284
-py2ls-0.2.4.33.dist-info/METADATA,sha256=YHx2Yww0F2wy0fPxBKC-pAkpLBq9db2D-McYtW7r8FI,20332
-py2ls-0.2.4.33.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-py2ls-0.2.4.33.dist-info/RECORD,,
+py2ls-0.2.4.34.dist-info/METADATA,sha256=KQ3NGQ07YFbGX__qttlwPsvwLBDPODKCJt9S7WStkqM,20332
+py2ls-0.2.4.34.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+py2ls-0.2.4.34.dist-info/RECORD,,