Spaces:
Sleeping
Sleeping
File size: 1,529 Bytes
884c49d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
import io
import os
import re
from setuptools import find_packages, setup
def get_long_description():
base_dir = os.path.abspath(os.path.dirname(__file__))
with io.open(os.path.join(base_dir, "README.md"), encoding="utf-8") as f:
return f.read()
def get_requirements():
with open("requirements.txt") as f:
return f.read().splitlines()
def get_version():
current_dir = os.path.abspath(os.path.dirname(__file__))
version_file = os.path.join(current_dir, "metaseg", "__init__.py")
with io.open(version_file, encoding="utf-8") as f:
return re.search(r'^__version__ = [\'"]([^\'"]*)[\'"]', f.read(), re.M).group(1)
_ALL_REQUIREMENTS = ["matplotlib", "pycocotools", "opencv-python", "onnx", "onnxruntime"]
_DEV_REQUIREMENTS = [
"black==23.*",
"isort==5.12.0",
"flake8",
"mypy",
]
extras = {
"all": _ALL_REQUIREMENTS,
"dev": _DEV_REQUIREMENTS,
}
setup(
name="metaseg",
license="Apache-2.0",
author="kadirnar",
long_description=get_long_description(),
long_description_content_type="text/markdown",
url="https://github.com/kadirnar/segment-anything-pip",
version=get_version(),
install_requires=get_requirements(),
packages=find_packages(exclude=("notebook")),
extras_require=extras,
python_requires=">=3.8",
)
|