代码拉取完成,页面将自动刷新
import os
import re
import platform
from setuptools import find_packages, setup
def get_version():
"""Get package version from app_info.py file"""
filename = "anylabeling/app_info.py"
with open(filename, encoding="utf-8") as f:
match = re.search(
r"""^__version__ = ['"]([^'"]*)['"]""", f.read(), re.M
)
if not match:
raise RuntimeError(f"{filename} doesn't contain __version__")
version = match.groups()[0]
return version
def get_preferred_device():
"""Get preferred device from app_info.py file: CPU or GPU"""
filename = "anylabeling/app_info.py"
with open(filename, encoding="utf-8") as f:
match = re.search(
r"""^__preferred_device__ = ['"]([^'"]*)['"]""", f.read(), re.M
)
if not match:
raise RuntimeError(f"{filename} doesn't contain __preferred_device__")
device = match.groups()[0]
return device
def get_package_name():
"""Get package name based on context"""
package_name = "anylabeling"
preferred_device = get_preferred_device()
if preferred_device == "GPU" and platform.system() != "Darwin":
package_name = "anylabeling-gpu"
return package_name
def get_install_requires():
"""Get python requirements based on context"""
install_requires = [
"imgviz>=0.11",
"natsort>=7.1.0",
"numpy",
"Pillow>=2.8",
"PyYAML",
"termcolor",
"opencv-python-headless",
'PyQt5>=5.15.7; platform_system != "Darwin"',
"onnx==1.13.1",
"qimage2ndarray==1.10.0",
"darkdetect==0.8.0",
]
# Add onnxruntime-gpu if GPU is preferred
# otherwise, add onnxruntime.
# Note: onnxruntime-gpu is not available on macOS
preferred_device = get_preferred_device()
if preferred_device == "GPU" and platform.system() != "Darwin":
install_requires.append("onnxruntime-gpu==1.14.1")
print("Building AnyLabeling with GPU support")
else:
install_requires.append("onnxruntime==1.14.1")
print("Building AnyLabeling without GPU support")
return install_requires
def get_long_description():
"""Read long description from README"""
with open("README.md", encoding="utf-8") as f:
long_description = f.read()
return long_description
setup(
name=get_package_name(),
version=get_version(),
packages=find_packages(),
description="Effortless data labeling with AI support",
long_description=get_long_description(),
long_description_content_type="text/markdown",
author="Viet-Anh Nguyen",
author_email="vietanh.dev@gmail.com",
url="https://github.com/vietanhdev/anylabeling",
install_requires=get_install_requires(),
license="GPLv3",
keywords="Image Annotation, Machine Learning, Deep Learning",
classifiers=[
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"Natural Language :: English",
"Operating System :: OS Independent",
"Programming Language :: Python",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3 :: Only",
],
include_package_data=True,
entry_points={
"console_scripts": [
"anylabeling=anylabeling.app:main",
],
},
)
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。