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

Added the Ipynb file for Google Colab.

Browse Source
This commit is contained in:
Arhenniuss 2025-05-13 14:43:56 +05:30 committed by GitHub
parent c817f5381c
commit 4f10c9115d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 319 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

319
rembg.ipynb Normal file
View File

@ -0,0 +1,319 @@
{
"nbformat": 4,
"nbformat_minor": 0,
"metadata": {
"colab": {
"provenance": [],
"gpuType": "T4"
},
"kernelspec": {
"name": "python3",
"display_name": "Python 3"
},
"language_info": {
"name": "python"
},
"accelerator": "GPU"
},
"cells": [
{
"cell_type": "code",
"execution_count": 3,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "hF9llNyHkiRB",
"outputId": "bd4e1cc0-f871-4c3f-d6e3-b503fe170f71"
},
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"Collecting rembg[cli,gpu]\n",
" Downloading rembg-2.0.65-py3-none-any.whl.metadata (20 kB)\n",
"Requirement already satisfied: jsonschema in /usr/local/lib/python3.11/dist-packages (from rembg[cli,gpu]) (4.23.0)\n",
"Requirement already satisfied: numpy in /usr/local/lib/python3.11/dist-packages (from rembg[cli,gpu]) (2.0.2)\n",
"Requirement already satisfied: opencv-python-headless in /usr/local/lib/python3.11/dist-packages (from rembg[cli,gpu]) (4.11.0.86)\n",
"Requirement already satisfied: pillow in /usr/local/lib/python3.11/dist-packages (from rembg[cli,gpu]) (11.2.1)\n",
"Requirement already satisfied: pooch in /usr/local/lib/python3.11/dist-packages (from rembg[cli,gpu]) (1.8.2)\n",
"Collecting pymatting (from rembg[cli,gpu])\n",
" Downloading PyMatting-1.1.13-py3-none-any.whl.metadata (7.5 kB)\n",
"Requirement already satisfied: scikit-image in /usr/local/lib/python3.11/dist-packages (from rembg[cli,gpu]) (0.25.2)\n",
"Requirement already satisfied: scipy in /usr/local/lib/python3.11/dist-packages (from rembg[cli,gpu]) (1.15.2)\n",
"Requirement already satisfied: tqdm in /usr/local/lib/python3.11/dist-packages (from rembg[cli,gpu]) (4.67.1)\n",
"Collecting onnxruntime-gpu (from rembg[cli,gpu])\n",
" Downloading onnxruntime_gpu-1.22.0-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl.metadata (4.9 kB)\n",
"Requirement already satisfied: aiohttp in /usr/local/lib/python3.11/dist-packages (from rembg[cli,gpu]) (3.11.15)\n",
"Collecting asyncer (from rembg[cli,gpu])\n",
" Downloading asyncer-0.0.8-py3-none-any.whl.metadata (6.7 kB)\n",
"Requirement already satisfied: click in /usr/local/lib/python3.11/dist-packages (from rembg[cli,gpu]) (8.1.8)\n",
"Collecting fastapi (from rembg[cli,gpu])\n",
" Downloading fastapi-0.115.12-py3-none-any.whl.metadata (27 kB)\n",
"Collecting filetype (from rembg[cli,gpu])\n",
" Downloading filetype-1.2.0-py2.py3-none-any.whl.metadata (6.5 kB)\n",
"Collecting gradio (from rembg[cli,gpu])\n",
" Downloading gradio-5.29.0-py3-none-any.whl.metadata (16 kB)\n",
"Collecting python-multipart (from rembg[cli,gpu])\n",
" Downloading python_multipart-0.0.20-py3-none-any.whl.metadata (1.8 kB)\n",
"Collecting uvicorn (from rembg[cli,gpu])\n",
" Downloading uvicorn-0.34.2-py3-none-any.whl.metadata (6.5 kB)\n",
"Collecting watchdog (from rembg[cli,gpu])\n",
" Downloading watchdog-6.0.0-py3-none-manylinux2014_x86_64.whl.metadata (44 kB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m44.3/44.3 kB\u001b[0m \u001b[31m4.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hRequirement already satisfied: aiohappyeyeballs>=2.3.0 in /usr/local/lib/python3.11/dist-packages (from aiohttp->rembg[cli,gpu]) (2.6.1)\n",
"Requirement already satisfied: aiosignal>=1.1.2 in /usr/local/lib/python3.11/dist-packages (from aiohttp->rembg[cli,gpu]) (1.3.2)\n",
"Requirement already satisfied: attrs>=17.3.0 in /usr/local/lib/python3.11/dist-packages (from aiohttp->rembg[cli,gpu]) (25.3.0)\n",
"Requirement already satisfied: frozenlist>=1.1.1 in /usr/local/lib/python3.11/dist-packages (from aiohttp->rembg[cli,gpu]) (1.6.0)\n",
"Requirement already satisfied: multidict<7.0,>=4.5 in /usr/local/lib/python3.11/dist-packages (from aiohttp->rembg[cli,gpu]) (6.4.3)\n",
"Requirement already satisfied: propcache>=0.2.0 in /usr/local/lib/python3.11/dist-packages (from aiohttp->rembg[cli,gpu]) (0.3.1)\n",
"Requirement already satisfied: yarl<2.0,>=1.17.0 in /usr/local/lib/python3.11/dist-packages (from aiohttp->rembg[cli,gpu]) (1.20.0)\n",
"Requirement already satisfied: anyio<5.0,>=3.4.0 in /usr/local/lib/python3.11/dist-packages (from asyncer->rembg[cli,gpu]) (4.9.0)\n",
"Collecting starlette<0.47.0,>=0.40.0 (from fastapi->rembg[cli,gpu])\n",
" Downloading starlette-0.46.2-py3-none-any.whl.metadata (6.2 kB)\n",
"Requirement already satisfied: pydantic!=1.8,!=1.8.1,!=2.0.0,!=2.0.1,!=2.1.0,<3.0.0,>=1.7.4 in /usr/local/lib/python3.11/dist-packages (from fastapi->rembg[cli,gpu]) (2.11.4)\n",
"Requirement already satisfied: typing-extensions>=4.8.0 in /usr/local/lib/python3.11/dist-packages (from fastapi->rembg[cli,gpu]) (4.13.2)\n",
"Collecting aiofiles<25.0,>=22.0 (from gradio->rembg[cli,gpu])\n",
" Downloading aiofiles-24.1.0-py3-none-any.whl.metadata (10 kB)\n",
"Collecting ffmpy (from gradio->rembg[cli,gpu])\n",
" Downloading ffmpy-0.5.0-py3-none-any.whl.metadata (3.0 kB)\n",
"Collecting gradio-client==1.10.0 (from gradio->rembg[cli,gpu])\n",
" Downloading gradio_client-1.10.0-py3-none-any.whl.metadata (7.1 kB)\n",
"Collecting groovy~=0.1 (from gradio->rembg[cli,gpu])\n",
" Downloading groovy-0.1.2-py3-none-any.whl.metadata (6.1 kB)\n",
"Requirement already satisfied: httpx>=0.24.1 in /usr/local/lib/python3.11/dist-packages (from gradio->rembg[cli,gpu]) (0.28.1)\n",
"Requirement already satisfied: huggingface-hub>=0.28.1 in /usr/local/lib/python3.11/dist-packages (from gradio->rembg[cli,gpu]) (0.31.1)\n",
"Requirement already satisfied: jinja2<4.0 in /usr/local/lib/python3.11/dist-packages (from gradio->rembg[cli,gpu]) (3.1.6)\n",
"Requirement already satisfied: markupsafe<4.0,>=2.0 in /usr/local/lib/python3.11/dist-packages (from gradio->rembg[cli,gpu]) (3.0.2)\n",
"Requirement already satisfied: orjson~=3.0 in /usr/local/lib/python3.11/dist-packages (from gradio->rembg[cli,gpu]) (3.10.18)\n",
"Requirement already satisfied: packaging in /usr/local/lib/python3.11/dist-packages (from gradio->rembg[cli,gpu]) (24.2)\n",
"Requirement already satisfied: pandas<3.0,>=1.0 in /usr/local/lib/python3.11/dist-packages (from gradio->rembg[cli,gpu]) (2.2.2)\n",
"Collecting pydub (from gradio->rembg[cli,gpu])\n",
" Downloading pydub-0.25.1-py2.py3-none-any.whl.metadata (1.4 kB)\n",
"Requirement already satisfied: pyyaml<7.0,>=5.0 in /usr/local/lib/python3.11/dist-packages (from gradio->rembg[cli,gpu]) (6.0.2)\n",
"Collecting ruff>=0.9.3 (from gradio->rembg[cli,gpu])\n",
" Downloading ruff-0.11.9-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (25 kB)\n",
"Collecting safehttpx<0.2.0,>=0.1.6 (from gradio->rembg[cli,gpu])\n",
" Downloading safehttpx-0.1.6-py3-none-any.whl.metadata (4.2 kB)\n",
"Collecting semantic-version~=2.0 (from gradio->rembg[cli,gpu])\n",
" Downloading semantic_version-2.10.0-py2.py3-none-any.whl.metadata (9.7 kB)\n",
"Collecting tomlkit<0.14.0,>=0.12.0 (from gradio->rembg[cli,gpu])\n",
" Downloading tomlkit-0.13.2-py3-none-any.whl.metadata (2.7 kB)\n",
"Requirement already satisfied: typer<1.0,>=0.12 in /usr/local/lib/python3.11/dist-packages (from gradio->rembg[cli,gpu]) (0.15.3)\n",
"Requirement already satisfied: fsspec in /usr/local/lib/python3.11/dist-packages (from gradio-client==1.10.0->gradio->rembg[cli,gpu]) (2025.3.2)\n",
"Requirement already satisfied: websockets<16.0,>=10.0 in /usr/local/lib/python3.11/dist-packages (from gradio-client==1.10.0->gradio->rembg[cli,gpu]) (15.0.1)\n",
"Requirement already satisfied: h11>=0.8 in /usr/local/lib/python3.11/dist-packages (from uvicorn->rembg[cli,gpu]) (0.16.0)\n",
"Requirement already satisfied: jsonschema-specifications>=2023.03.6 in /usr/local/lib/python3.11/dist-packages (from jsonschema->rembg[cli,gpu]) (2025.4.1)\n",
"Requirement already satisfied: referencing>=0.28.4 in /usr/local/lib/python3.11/dist-packages (from jsonschema->rembg[cli,gpu]) (0.36.2)\n",
"Requirement already satisfied: rpds-py>=0.7.1 in /usr/local/lib/python3.11/dist-packages (from jsonschema->rembg[cli,gpu]) (0.24.0)\n",
"Collecting coloredlogs (from onnxruntime-gpu->rembg[cli,gpu])\n",
" Downloading coloredlogs-15.0.1-py2.py3-none-any.whl.metadata (12 kB)\n",
"Requirement already satisfied: flatbuffers in /usr/local/lib/python3.11/dist-packages (from onnxruntime-gpu->rembg[cli,gpu]) (25.2.10)\n",
"Requirement already satisfied: protobuf in /usr/local/lib/python3.11/dist-packages (from onnxruntime-gpu->rembg[cli,gpu]) (5.29.4)\n",
"Requirement already satisfied: sympy in /usr/local/lib/python3.11/dist-packages (from onnxruntime-gpu->rembg[cli,gpu]) (1.13.1)\n",
"Requirement already satisfied: platformdirs>=2.5.0 in /usr/local/lib/python3.11/dist-packages (from pooch->rembg[cli,gpu]) (4.3.8)\n",
"Requirement already satisfied: requests>=2.19.0 in /usr/local/lib/python3.11/dist-packages (from pooch->rembg[cli,gpu]) (2.32.3)\n",
"Requirement already satisfied: numba!=0.49.0 in /usr/local/lib/python3.11/dist-packages (from pymatting->rembg[cli,gpu]) (0.60.0)\n",
"Requirement already satisfied: networkx>=3.0 in /usr/local/lib/python3.11/dist-packages (from scikit-image->rembg[cli,gpu]) (3.4.2)\n",
"Requirement already satisfied: imageio!=2.35.0,>=2.33 in /usr/local/lib/python3.11/dist-packages (from scikit-image->rembg[cli,gpu]) (2.37.0)\n",
"Requirement already satisfied: tifffile>=2022.8.12 in /usr/local/lib/python3.11/dist-packages (from scikit-image->rembg[cli,gpu]) (2025.3.30)\n",
"Requirement already satisfied: lazy-loader>=0.4 in /usr/local/lib/python3.11/dist-packages (from scikit-image->rembg[cli,gpu]) (0.4)\n",
"Requirement already satisfied: idna>=2.8 in /usr/local/lib/python3.11/dist-packages (from anyio<5.0,>=3.4.0->asyncer->rembg[cli,gpu]) (3.10)\n",
"Requirement already satisfied: sniffio>=1.1 in /usr/local/lib/python3.11/dist-packages (from anyio<5.0,>=3.4.0->asyncer->rembg[cli,gpu]) (1.3.1)\n",
"Requirement already satisfied: certifi in /usr/local/lib/python3.11/dist-packages (from httpx>=0.24.1->gradio->rembg[cli,gpu]) (2025.4.26)\n",
"Requirement already satisfied: httpcore==1.* in /usr/local/lib/python3.11/dist-packages (from httpx>=0.24.1->gradio->rembg[cli,gpu]) (1.0.9)\n",
"Requirement already satisfied: filelock in /usr/local/lib/python3.11/dist-packages (from huggingface-hub>=0.28.1->gradio->rembg[cli,gpu]) (3.18.0)\n",
"Requirement already satisfied: hf-xet<2.0.0,>=1.1.0 in /usr/local/lib/python3.11/dist-packages (from huggingface-hub>=0.28.1->gradio->rembg[cli,gpu]) (1.1.0)\n",
"Requirement already satisfied: llvmlite<0.44,>=0.43.0dev0 in /usr/local/lib/python3.11/dist-packages (from numba!=0.49.0->pymatting->rembg[cli,gpu]) (0.43.0)\n",
"Requirement already satisfied: python-dateutil>=2.8.2 in /usr/local/lib/python3.11/dist-packages (from pandas<3.0,>=1.0->gradio->rembg[cli,gpu]) (2.9.0.post0)\n",
"Requirement already satisfied: pytz>=2020.1 in /usr/local/lib/python3.11/dist-packages (from pandas<3.0,>=1.0->gradio->rembg[cli,gpu]) (2025.2)\n",
"Requirement already satisfied: tzdata>=2022.7 in /usr/local/lib/python3.11/dist-packages (from pandas<3.0,>=1.0->gradio->rembg[cli,gpu]) (2025.2)\n",
"Requirement already satisfied: annotated-types>=0.6.0 in /usr/local/lib/python3.11/dist-packages (from pydantic!=1.8,!=1.8.1,!=2.0.0,!=2.0.1,!=2.1.0,<3.0.0,>=1.7.4->fastapi->rembg[cli,gpu]) (0.7.0)\n",
"Requirement already satisfied: pydantic-core==2.33.2 in /usr/local/lib/python3.11/dist-packages (from pydantic!=1.8,!=1.8.1,!=2.0.0,!=2.0.1,!=2.1.0,<3.0.0,>=1.7.4->fastapi->rembg[cli,gpu]) (2.33.2)\n",
"Requirement already satisfied: typing-inspection>=0.4.0 in /usr/local/lib/python3.11/dist-packages (from pydantic!=1.8,!=1.8.1,!=2.0.0,!=2.0.1,!=2.1.0,<3.0.0,>=1.7.4->fastapi->rembg[cli,gpu]) (0.4.0)\n",
"Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.11/dist-packages (from requests>=2.19.0->pooch->rembg[cli,gpu]) (3.4.2)\n",
"Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.11/dist-packages (from requests>=2.19.0->pooch->rembg[cli,gpu]) (2.4.0)\n",
"Requirement already satisfied: shellingham>=1.3.0 in /usr/local/lib/python3.11/dist-packages (from typer<1.0,>=0.12->gradio->rembg[cli,gpu]) (1.5.4)\n",
"Requirement already satisfied: rich>=10.11.0 in /usr/local/lib/python3.11/dist-packages (from typer<1.0,>=0.12->gradio->rembg[cli,gpu]) (13.9.4)\n",
"Collecting humanfriendly>=9.1 (from coloredlogs->onnxruntime-gpu->rembg[cli,gpu])\n",
" Downloading humanfriendly-10.0-py2.py3-none-any.whl.metadata (9.2 kB)\n",
"Requirement already satisfied: mpmath<1.4,>=1.1.0 in /usr/local/lib/python3.11/dist-packages (from sympy->onnxruntime-gpu->rembg[cli,gpu]) (1.3.0)\n",
"Requirement already satisfied: six>=1.5 in /usr/local/lib/python3.11/dist-packages (from python-dateutil>=2.8.2->pandas<3.0,>=1.0->gradio->rembg[cli,gpu]) (1.17.0)\n",
"Requirement already satisfied: markdown-it-py>=2.2.0 in /usr/local/lib/python3.11/dist-packages (from rich>=10.11.0->typer<1.0,>=0.12->gradio->rembg[cli,gpu]) (3.0.0)\n",
"Requirement already satisfied: pygments<3.0.0,>=2.13.0 in /usr/local/lib/python3.11/dist-packages (from rich>=10.11.0->typer<1.0,>=0.12->gradio->rembg[cli,gpu]) (2.19.1)\n",
"Requirement already satisfied: mdurl~=0.1 in /usr/local/lib/python3.11/dist-packages (from markdown-it-py>=2.2.0->rich>=10.11.0->typer<1.0,>=0.12->gradio->rembg[cli,gpu]) (0.1.2)\n",
"Downloading asyncer-0.0.8-py3-none-any.whl (9.2 kB)\n",
"Downloading fastapi-0.115.12-py3-none-any.whl (95 kB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m95.2/95.2 kB\u001b[0m \u001b[31m9.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hDownloading filetype-1.2.0-py2.py3-none-any.whl (19 kB)\n",
"Downloading gradio-5.29.0-py3-none-any.whl (54.1 MB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m54.1/54.1 MB\u001b[0m \u001b[31m18.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hDownloading gradio_client-1.10.0-py3-none-any.whl (322 kB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m322.9/322.9 kB\u001b[0m \u001b[31m26.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hDownloading python_multipart-0.0.20-py3-none-any.whl (24 kB)\n",
"Downloading uvicorn-0.34.2-py3-none-any.whl (62 kB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m62.5/62.5 kB\u001b[0m \u001b[31m6.0 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hDownloading onnxruntime_gpu-1.22.0-cp311-cp311-manylinux_2_27_x86_64.manylinux_2_28_x86_64.whl (283.2 MB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m283.2/283.2 MB\u001b[0m \u001b[31m6.8 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hDownloading PyMatting-1.1.13-py3-none-any.whl (54 kB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m54.5/54.5 kB\u001b[0m \u001b[31m5.1 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hDownloading rembg-2.0.65-py3-none-any.whl (41 kB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m41.2/41.2 kB\u001b[0m \u001b[31m3.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hDownloading watchdog-6.0.0-py3-none-manylinux2014_x86_64.whl (79 kB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m79.1/79.1 kB\u001b[0m \u001b[31m8.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hDownloading aiofiles-24.1.0-py3-none-any.whl (15 kB)\n",
"Downloading groovy-0.1.2-py3-none-any.whl (14 kB)\n",
"Downloading ruff-0.11.9-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (11.5 MB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m11.5/11.5 MB\u001b[0m \u001b[31m115.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hDownloading safehttpx-0.1.6-py3-none-any.whl (8.7 kB)\n",
"Downloading semantic_version-2.10.0-py2.py3-none-any.whl (15 kB)\n",
"Downloading starlette-0.46.2-py3-none-any.whl (72 kB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m72.0/72.0 kB\u001b[0m \u001b[31m6.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hDownloading tomlkit-0.13.2-py3-none-any.whl (37 kB)\n",
"Downloading coloredlogs-15.0.1-py2.py3-none-any.whl (46 kB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m46.0/46.0 kB\u001b[0m \u001b[31m4.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hDownloading ffmpy-0.5.0-py3-none-any.whl (6.0 kB)\n",
"Downloading pydub-0.25.1-py2.py3-none-any.whl (32 kB)\n",
"Downloading humanfriendly-10.0-py2.py3-none-any.whl (86 kB)\n",
"\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m86.8/86.8 kB\u001b[0m \u001b[31m8.7 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
"\u001b[?25hInstalling collected packages: pydub, filetype, watchdog, uvicorn, tomlkit, semantic-version, ruff, python-multipart, humanfriendly, groovy, ffmpy, aiofiles, starlette, pymatting, coloredlogs, asyncer, safehttpx, onnxruntime-gpu, gradio-client, fastapi, rembg, gradio\n",
"Successfully installed aiofiles-24.1.0 asyncer-0.0.8 coloredlogs-15.0.1 fastapi-0.115.12 ffmpy-0.5.0 filetype-1.2.0 gradio-5.29.0 gradio-client-1.10.0 groovy-0.1.2 humanfriendly-10.0 onnxruntime-gpu-1.22.0 pydub-0.25.1 pymatting-1.1.13 python-multipart-0.0.20 rembg-2.0.65 ruff-0.11.9 safehttpx-0.1.6 semantic-version-2.10.0 starlette-0.46.2 tomlkit-0.13.2 uvicorn-0.34.2 watchdog-6.0.0\n",
"Cloning into 'rembg'...\n",
"remote: Enumerating objects: 2281, done.\u001b[K\n",
"remote: Counting objects: 100% (918/918), done.\u001b[K\n",
"remote: Compressing objects: 100% (318/318), done.\u001b[K\n",
"remote: Total 2281 (delta 759), reused 600 (delta 600), pack-reused 1363 (from 3)\u001b[K\n",
"Receiving objects: 100% (2281/2281), 71.38 MiB | 16.89 MiB/s, done.\n",
"Resolving deltas: 100% (1349/1349), done.\n"
]
}
],
"source": [
"! pip install \"rembg[gpu,cli]\"\n",
"! git clone https://github.com/danielgatis/rembg.git"
]
},
{
"cell_type": "code",
"source": [
"cd rembg/"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "Eu-NvAFeyKIX",
"outputId": "d7cce54f-e3b7-4879-8313-c23c432d324b"
},
"execution_count": 4,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"/content/rembg\n"
]
}
]
},
{
"cell_type": "code",
"source": [
"! wget https://huggingface.co/spaces/KenjieDec/RemBG/resolve/main/app.py\n",
"! sed -i 's/app.launch()/app.launch(share=True)/' app.py"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "pHhQCQvDy2yv",
"outputId": "96c28494-4bcf-4574-dccc-5345f28ec5e5"
},
"execution_count": 5,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"--2025-05-13 08:38:54-- https://huggingface.co/spaces/KenjieDec/RemBG/resolve/main/app.py\n",
"Resolving huggingface.co (huggingface.co)... 13.35.202.40, 13.35.202.97, 13.35.202.121, ...\n",
"Connecting to huggingface.co (huggingface.co)|13.35.202.40|:443... connected.\n",
"HTTP request sent, awaiting response... 200 OK\n",
"Length: 3525 (3.4K) [text/plain]\n",
"Saving to: ‘app.py’\n",
"\n",
"app.py 100%[===================>] 3.44K --.-KB/s in 0s \n",
"\n",
"2025-05-13 08:38:54 (1.96 GB/s) - ‘app.py’ saved [3525/3525]\n",
"\n"
]
}
]
},
{
"cell_type": "code",
"source": [
"!python app.py --share"
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "exMGiaIruPIp",
"outputId": "660061fd-7df9-4516-cae9-83194c16e50c"
},
"execution_count": null,
"outputs": [
{
"output_type": "stream",
"name": "stdout",
"text": [
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-general-epoch_244.onnx' to file '/root/.u2net/birefnet-general.onnx'.\n",
"100%|███████████████████████████████████████| 973M/973M [00:00<00:00, 5.26TB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-general-bb_swin_v1_tiny-epoch_232.onnx' to file '/root/.u2net/birefnet-general-lite.onnx'.\n",
"100%|███████████████████████████████████████| 224M/224M [00:00<00:00, 1.33TB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-portrait-epoch_150.onnx' to file '/root/.u2net/birefnet-portrait.onnx'.\n",
"100%|███████████████████████████████████████| 973M/973M [00:00<00:00, 5.83TB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-DIS-epoch_590.onnx' to file '/root/.u2net/birefnet-dis.onnx'.\n",
"100%|███████████████████████████████████████| 973M/973M [00:00<00:00, 5.01TB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-HRSOD_DHU-epoch_115.onnx' to file '/root/.u2net/birefnet-hrsod.onnx'.\n",
"100%|███████████████████████████████████████| 973M/973M [00:00<00:00, 4.18TB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-COD-epoch_125.onnx' to file '/root/.u2net/birefnet-cod.onnx'.\n",
"100%|███████████████████████████████████████| 973M/973M [00:00<00:00, 4.87TB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/BiRefNet-massive-TR_DIS5K_TR_TEs-epoch_420.onnx' to file '/root/.u2net/birefnet-massive.onnx'.\n",
"100%|███████████████████████████████████████| 973M/973M [00:00<00:00, 5.02TB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/isnet-anime.onnx' to file '/root/.u2net/isnet-anime.onnx'.\n",
"100%|████████████████████████████████████████| 176M/176M [00:00<00:00, 678GB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/isnet-general-use.onnx' to file '/root/.u2net/isnet-general-use.onnx'.\n",
"100%|████████████████████████████████████████| 179M/179M [00:00<00:00, 938GB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/sam_vit_b_01ec64.encoder.onnx' to file '/root/.u2net/sam_vit_b_01ec64.encoder.onnx'.\n",
"100%|███████████████████████████████████████| 359M/359M [00:00<00:00, 1.66TB/s]\n",
"SHA256 hash of downloaded file: 62a29c39ebc8ae4dfc3d9ec8eae28c28efc8360233fa8ffba229eb9f6ac0c30a\n",
"Use this value as the 'known_hash' argument of 'pooch.retrieve' to ensure that the file hasn't changed if it is downloaded again in the future.\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/sam_vit_b_01ec64.decoder.onnx' to file '/root/.u2net/sam_vit_b_01ec64.decoder.onnx'.\n",
"100%|█████████████████████████████████████| 16.5M/16.5M [00:00<00:00, 86.6GB/s]\n",
"SHA256 hash of downloaded file: 9820a48a2d002aead7b4dab204d813176a810a5f1dc8f32e32f759cc070ae97e\n",
"Use this value as the 'known_hash' argument of 'pooch.retrieve' to ensure that the file hasn't changed if it is downloaded again in the future.\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/silueta.onnx' to file '/root/.u2net/silueta.onnx'.\n",
"100%|██████████████████████████████████████| 44.2M/44.2M [00:00<00:00, 237GB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_cloth_seg.onnx' to file '/root/.u2net/u2net_cloth_seg.onnx'.\n",
"100%|████████████████████████████████████████| 176M/176M [00:00<00:00, 915GB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_human_seg.onnx' to file '/root/.u2net/u2net_human_seg.onnx'.\n",
"100%|███████████████████████████████████████| 176M/176M [00:00<00:00, 1.05TB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx' to file '/root/.u2net/u2net.onnx'.\n",
"100%|███████████████████████████████████████| 176M/176M [00:00<00:00, 1.07TB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2netp.onnx' to file '/root/.u2net/u2netp.onnx'.\n",
"100%|█████████████████████████████████████| 4.57M/4.57M [00:00<00:00, 25.4GB/s]\n",
"Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/bria-rmbg-2.0.onnx' to file '/root/.u2net/bria-rmbg.onnx'.\n",
"100%|█████████████████████████████████████| 1.02G/1.02G [00:00<00:00, 4.92TB/s]\n",
"* Running on local URL: http://127.0.0.1:7860\n",
"* Running on public URL: https://3b40186f4fbbdaaf10.gradio.live\n",
"\n",
"This share link expires in 1 week. For free permanent hosting and GPU upgrades, run `gradio deploy` from the terminal in the working directory to deploy to Hugging Face Spaces (https://huggingface.co/spaces)\n"
]
}
]
}
]
}