Initial Commit

.dockerignore

@@ -0,0 +1,9 @@
+*
+!rembg
+!setup.py
+!setup.cfg
+!requirements.txt
+!requirements-cpu.txt
+!requirements-gpu.txt
+!versioneer.py
+!README.md

.editorconfig

@@ -0,0 +1,11 @@
+# https://editorconfig.org/
+
+root = true
+
+[*]
+indent_style = space
+indent_size = 4
+insert_final_newline = true
+trim_trailing_whitespace = true
+end_of_line = lf
+charset = utf-8

.gitattributes

@@ -0,0 +1,2 @@
+rembg/_version.py export-subst
+*.onnx filter=lfs diff=lfs merge=lfs -text

.github/FUNDING.yml

@@ -0,0 +1,2 @@
+github: [danielgatis]
+custom: ["https://www.buymeacoffee.com/danielgatis"]

.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,33 @@
+---
+name: Bug report
+about: Create a report to help us improve
+title: "[BUG] ..."
+labels: bug
+assignees: ""
+---
+
+**Describe the bug**
+A clear and concise description of what the bug is.
+
+**To Reproduce**
+Steps to reproduce the behavior:
+
+1. Go to '...'
+2. Click on '....'
+3. Scroll down to '....'
+4. See error
+
+**Expected behavior**
+A clear and concise description of what you expected to happen.
+
+**Images**
+Input images to reproduce.
+
+**OS Version:**
+iOS 22
+
+**Rembg version:**
+v2.0.21
+
+**Additional context**
+Add any other context about the problem here.

.github/ISSUE_TEMPLATE/feature_request.md

@@ -0,0 +1,19 @@
+---
+name: Feature request
+about: Suggest an idea for this project
+title: "[FEATURE] ..."
+labels: enhancement
+assignees: ""
+---
+
+**Is your feature request related to a problem? Please describe.**
+A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
+
+**Describe the solution you'd like**
+A clear and concise description of what you want to happen.
+
+**Describe alternatives you've considered**
+A clear and concise description of any alternative solutions or features you've considered.
+
+**Additional context**
+Add any other context or screenshots about the feature request here.

.github/workflows/close_inactive_issues.yml

@@ -0,0 +1,23 @@
+name: Close inactive issues
+
+on:
+    schedule:
+        - cron: "30 1 * * *"
+
+jobs:
+    close-issues:
+        runs-on: ubuntu-latest
+        permissions:
+            issues: write
+            pull-requests: write
+        steps:
+            - uses: actions/stale@v5
+              with:
+                  days-before-issue-stale: 30
+                  days-before-issue-close: 14
+                  stale-issue-label: "stale"
+                  stale-issue-message: "This issue is stale because it has been open for 30 days with no activity."
+                  close-issue-message: "This issue was closed because it has been inactive for 14 days since being marked as stale."
+                  days-before-pr-stale: -1
+                  days-before-pr-close: -1
+                  repo-token: ${{ secrets.GITHUB_TOKEN }}

.github/workflows/lint_python.yml

@@ -0,0 +1,18 @@
+name: Lint
+
+on: [pull_request, push]
+
+jobs:
+    lint_python:
+        runs-on: ubuntu-latest
+        steps:
+            - uses: actions/checkout@v2
+            - uses: actions/setup-python@v2
+            - run: pip install --upgrade pip wheel
+            - run: pip install bandit black flake8 flake8-bugbear flake8-comprehensions isort safety mypy
+            - run: mypy --install-types --non-interactive --ignore-missing-imports ./rembg
+            - run: bandit --recursive --skip B101,B104,B310,B311,B303 --exclude ./rembg/_version.py ./rembg
+            - run: black --force-exclude rembg/_version.py --check --diff ./rembg
+            - run: flake8 ./rembg --count --ignore=B008,C901,E203,E266,E731,F401,F811,F841,W503 --max-line-length=120 --show-source --statistics --exclude ./rembg/_version.py
+            - run: isort --check-only --profile black ./rembg
+            - run: safety check

.github/workflows/publish_docker.yml

@@ -0,0 +1,28 @@
+name: Publish Docker image
+
+on:
+    push:
+        tags:
+            - "v*.*.*"
+
+jobs:
+    push_to_registry:
+        name: Push Docker image to Docker Hub
+        runs-on: ubuntu-latest
+        steps:
+            - name: Checkout
+              uses: actions/checkout@v2
+            - name: Login to Docker Hub
+              uses: docker/login-action@v1
+              with:
+                  username: ${{ secrets.DOCKER_HUB_USERNAME }}
+                  password: ${{ secrets.DOCKER_HUB_ACCESS_TOKEN }}
+            - name: Set up Docker Buildx
+              uses: docker/setup-buildx-action@v1
+            - name: Build and push
+              uses: docker/build-push-action@v2
+              with:
+                  context: .
+                  file: ./Dockerfile
+                  push: true
+                  tags: ${{ secrets.DOCKER_HUB_USERNAME }}/rembg-aws-lambda:latest

.github/workflows/publish_pypi.yml

@@ -0,0 +1,28 @@
+name: Publish to Pypi
+
+on:
+    push:
+        tags:
+            - "v*.*.*"
+
+jobs:
+    push_to_pypi:
+        runs-on: ubuntu-latest
+        steps:
+            - uses: actions/checkout@v2
+            - uses: actions/setup-python@v2
+              with:
+                  python-version: 3.9
+
+            - name: "Installs dependencies"
+              run: |
+                  python3 -m pip install --upgrade pip
+                  python3 -m pip install setuptools wheel twine
+
+            - name: "Builds and uploads to PyPI"
+              run: |
+                  python3 setup.py sdist bdist_wheel
+                  python3 -m twine upload dist/*
+              env:
+                  TWINE_USERNAME: __token__
+                  TWINE_PASSWORD: ${{ secrets.PYPI_API_TOKEN }}

.github/workflows/test-install.yml

@@ -0,0 +1,25 @@
+name: Test installation
+
+on: [push]
+
+jobs:
+    build:
+        runs-on: ubuntu-latest
+        strategy:
+            matrix:
+                python-version: ["3.8", "3.9", "3.10"]
+
+        steps:
+            - uses: actions/checkout@v3
+            - name: Set up Python ${{ matrix.python-version }}
+              uses: actions/setup-python@v4
+              with:
+                  python-version: ${{ matrix.python-version }}
+            - name: Install package
+              run: |
+                  python -m pip install --upgrade pip
+                  pip install .
+                  pip install pytest
+            - name: Test installation with pytest
+              run: |
+                  pytest

.github/workflows/test-specific-environment.yml

@@ -0,0 +1,25 @@
+name: Test in specific environment
+
+on: [push]
+
+jobs:
+    build:
+        runs-on: ubuntu-latest
+        strategy:
+            matrix:
+                python-version: ["3.8", "3.9", "3.10"]
+
+        steps:
+            - uses: actions/checkout@v3
+            - name: Set up Python ${{ matrix.python-version }}
+              uses: actions/setup-python@v4
+              with:
+                  python-version: ${{ matrix.python-version }}
+            - name: Set up environment
+              run: |
+                  python -m pip install --upgrade pip
+                  pip install pytest
+                  pip install -r requirements.txt
+            - name: Test with pytest
+              run: |
+                  PYTHONPATH=$PYTHONPATH:. pytest .

.gitignore

@@ -0,0 +1,21 @@
+# general things to ignore
+build/
+dist/
+.venv/
+.direnv/
+*.spec
+*.egg-info/
+*.egg
+*.py[cod]
+__pycache__/
+*.so
+*~≈
+.envrc
+.python-version
+.idea
+.pytest_cache
+
+# due to using tox and pytest
+.tox
+.cache
+.mypy_cache

Dockerfile

@@ -0,0 +1,31 @@
+FROM nvidia/cuda:11.6.0-runtime-ubuntu18.04
+
+ENV DEBIAN_FRONTEND noninteractive
+
+RUN rm /etc/apt/sources.list.d/cuda.list || true
+RUN rm /etc/apt/sources.list.d/nvidia-ml.list || true
+RUN apt-key del 7fa2af80
+RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
+RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
+
+RUN apt update -y
+RUN apt upgrade -y
+RUN apt install -y curl wget software-properties-common
+RUN add-apt-repository ppa:deadsnakes/ppa
+RUN apt install -y python3.9 python3.9-distutils
+RUN curl https://bootstrap.pypa.io/get-pip.py | python3.9
+
+WORKDIR /rembg
+
+COPY . .
+RUN python3.9 -m pip install .[gpu]
+
+RUN mkdir -p ~/.u2net
+RUN wget https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2netp.onnx -O ~/.u2net/u2netp.onnx
+RUN wget https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx -O ~/.u2net/u2net.onnx
+RUN wget https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_human_seg.onnx -O ~/.u2net/u2net_human_seg.onnx
+RUN wget https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_cloth_seg.onnx -O ~/.u2net/u2net_cloth_seg.onnx
+
+EXPOSE 5000
+ENTRYPOINT ["rembg"]
+CMD ["--help"]

LICENSE.txt

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2020 Daniel Gatis
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

MANIFEST.in

@@ -0,0 +1,11 @@
+include MANIFEST.in
+include LICENSE.txt
+include README.md
+include setup.py
+include pyproject.toml
+include requirements.txt
+include requirements-gpu.txt
+include rembg/*.onnx
+
+include versioneer.py
+include rembg/_version.py