mirror of
https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg
synced 2025-07-30 08:02:00 +08:00
93 lines
2.2 KiB
Python
93 lines
2.2 KiB
Python
import os
|
|
import pathlib
|
|
import sys
|
|
|
|
sys.path.append(os.path.dirname(__file__))
|
|
from setuptools import find_packages, setup
|
|
|
|
import versioneer
|
|
|
|
here = pathlib.Path(__file__).parent.resolve()
|
|
|
|
long_description = (here / "README.md").read_text(encoding="utf-8")
|
|
|
|
install_requires = [
|
|
"jsonschema",
|
|
"numpy",
|
|
"opencv-python-headless",
|
|
"pillow",
|
|
"pooch",
|
|
"pymatting",
|
|
"scikit-image",
|
|
"scipy",
|
|
"tqdm",
|
|
]
|
|
|
|
extras_require = {
|
|
"dev": [
|
|
"bandit",
|
|
"black",
|
|
"flake8",
|
|
"imagehash",
|
|
"isort",
|
|
"mypy",
|
|
"pytest",
|
|
"setuptools",
|
|
"twine",
|
|
"wheel",
|
|
],
|
|
"cpu": ["onnxruntime"],
|
|
"gpu": ["onnxruntime-gpu"],
|
|
"cli": [
|
|
"aiohttp",
|
|
"asyncer",
|
|
"click",
|
|
"fastapi",
|
|
"filetype",
|
|
"gradio",
|
|
"python-multipart",
|
|
"uvicorn",
|
|
"watchdog",
|
|
],
|
|
}
|
|
|
|
entry_points = {
|
|
"console_scripts": [
|
|
"rembg=rembg.cli:main",
|
|
],
|
|
}
|
|
|
|
|
|
setup(
|
|
name="rembg",
|
|
description="Remove image background",
|
|
long_description=long_description,
|
|
long_description_content_type="text/markdown",
|
|
url="https://github.com/danielgatis/rembg",
|
|
author="Daniel Gatis",
|
|
author_email="danielgatis@gmail.com",
|
|
classifiers=[
|
|
"License :: OSI Approved :: MIT License",
|
|
"Topic :: Scientific/Engineering",
|
|
"Topic :: Scientific/Engineering :: Mathematics",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
"Topic :: Software Development",
|
|
"Topic :: Software Development :: Libraries",
|
|
"Topic :: Software Development :: Libraries :: Python Modules",
|
|
"Programming Language :: Python",
|
|
"Programming Language :: Python :: 3 :: Only",
|
|
"Programming Language :: Python :: 3.10",
|
|
"Programming Language :: Python :: 3.11",
|
|
"Programming Language :: Python :: 3.12",
|
|
"Programming Language :: Python :: 3.13",
|
|
],
|
|
keywords="remove, background, u2net",
|
|
python_requires=">=3.10, <3.14",
|
|
packages=find_packages(),
|
|
install_requires=install_requires,
|
|
entry_points=entry_points,
|
|
extras_require=extras_require,
|
|
version=versioneer.get_version(),
|
|
cmdclass=versioneer.get_cmdclass(),
|
|
)
|