Daniel Gatis 4 年之前
父节点
当前提交
63f52a3a6a
共有 2 个文件被更改,包括 4 次插入2 次删除
  1. 1 1
      setup.py
  2. 3 1
      src/rembg/cmd/cli.py

+ 1 - 1
setup.py

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

+ 3 - 1
src/rembg/cmd/cli.py

@@ -100,7 +100,9 @@ def main():
             if os.path.isfile(path):
             if os.path.isfile(path):
                 files.add(path)
                 files.add(path)
             else:
             else:
-                full_paths += glob.glob(path + "/*")
+                full_paths += set(glob.glob(path + "/*")) - set(
+                    glob.glob(path + "/*.out.png")
+                )
 
 
         for fi in tqdm(files):
         for fi in tqdm(files):
             fi_type = filetype.guess(fi)
             fi_type = filetype.guess(fi)