AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-16 08:56:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

update versions

Browse Source
This commit is contained in:
Daniel Gatis 2020-11-09 14:53:35 -03:00
parent 8d5e4aba5d
commit b6cfd4b045
3 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
requirements.txt
View File

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

4
setup.py
View File

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

2
src/rembg/bg.py
View File

@ -64,7 +64,7 @@ def naive_cutout(img, mask):
return cutout return cutout
@functools.lru_cache @functools.lru_cache(maxsize=None)
def get_model(model_name): def get_model(model_name):
if model_name == "u2netp": if model_name == "u2netp":
return detect.load_model(model_name="u2netp") return detect.load_model(model_name="u2netp")