Daniel Gatis пре 4 година
родитељ
комит
b6cfd4b045
3 измењених фајлова са 10 додато и 10 уклоњено
  1. 7 7
      requirements.txt
  2. 2 2
      setup.py
  3. 1 1
      src/rembg/bg.py

+ 7 - 7
requirements.txt

@@ -1,11 +1,11 @@
 flask==1.1.2
-numpy==1.19.1
-pillow==7.2.0
+numpy==1.19.4
+pillow==8.0.1
 scikit-image==0.17.2
-torch==1.6.0
-torchvision==0.7.0
+torch==1.7.0
+torchvision==0.8.1
 waitress==1.4.4
-tqdm==4.48.2
+tqdm==4.51.0
 requests==2.24.0
-scipy==1.5.2
-pymatting==1.0.6
+scipy==1.5.4
+pymatting==1.1.1

+ 2 - 2
setup.py

@@ -11,7 +11,7 @@ with open("requirements.txt") as f:
 
 setup(
     name="rembg",
-    version="1.0.14",
+    version="1.0.15",
     description="Remove image background",
     long_description=long_description,
     long_description_content_type="text/markdown",
@@ -25,7 +25,7 @@ setup(
     keywords="remove, background, u2net",
     package_dir={"": "src"},
     packages=find_packages(where="src"),
-    python_requires=">=3.5, <4",
+    python_requires=">=3.8, <4",
     install_requires=requireds,
     entry_points={
         "console_scripts": [

+ 1 - 1
src/rembg/bg.py

@@ -64,7 +64,7 @@ def naive_cutout(img, mask):
     return cutout
 
 
[email protected]_cache
[email protected]_cache(maxsize=None)
 def get_model(model_name):
     if model_name == "u2netp":
         return detect.load_model(model_name="u2netp")