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

fix typo

Browse Source
This commit is contained in:
Daniel Gatis 2020-11-13 13:47:24 -03:00
parent 4659d52273
commit 332c16b58e
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -11,7 +11,7 @@ with open("requirements.txt") as f:
setup( setup(
name="rembg", name="rembg",
version="1.0.15", version="1.0.16",
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",

6
src/rembg/cmd/cli.py
View File

@ -3,6 +3,7 @@ import glob
import os import os
import filetype import filetype
from distutils.util import strtobool from distutils.util import strtobool
from tqdm import tqdm
from ..bg import remove from ..bg import remove
@ -89,11 +90,12 @@ def main():
else: else:
full_paths += glob.glob(path + "/*") full_paths += glob.glob(path + "/*")
for fi in files: for fi in tqdm(files):
fi_type = filetype.guess(fi) fi_type = filetype.guess(fi)
if fi_type is None: if fi_type is None:
continue continue
elif if_type.mime.find('image') < 0: elif fi_type.mime.find('image') < 0:
continue continue
with open(fi, "rb") as input: with open(fi, "rb") as input: