Merge pull request #200 from DefTruth/dev

feat(pylitex): init pylitex setup configs
DefTruth · Feb 20, 2022 · e0fac36 · e0fac36
2 parents f2cc064 + 9a6627d
commit e0fac36
Show file tree

Hide file tree

Showing 29 changed files with 63 additions and 5 deletions.
diff --git a/.gitignore b/.gitignore
@@ -4,6 +4,4 @@ cmake-build-debug
 cmake-build-release
 .idea
 build-debug.sh
-python
-command
-
+__pycache__
diff --git a/README.md b/README.md
@@ -1084,4 +1084,4 @@ auto *nanodet = new lite::ncnn::cv::detection::NanoDet(param_path, bin_path);
 ## 11. Contribute
 <div id="lite.ai.toolkit-Contribute"></div>  
 
-How to add your own models and become a contributor? For specific steps, please refer to [CONTRIBUTING.zh.md](docs/contrib/CONTRIBUTING.zh.md), or if you like this project please ❤️ consider ⭐️🌟 star this repo, as it is the simplest way to support me.
+How to add your own models and become a contributor? For specific steps, please refer to [CONTRIBUTING.zh.md](https://github.com/DefTruth/lite.ai.toolkit/issues/191), or if you like this project please ❤️ consider ⭐️🌟 star this repo, as it is the simplest way to support me.
diff --git a/README.zh.md b/README.zh.md
@@ -1082,5 +1082,5 @@ auto *nanodet = new lite::ncnn::cv::detection::NanoDet(param_path, bin_path);
 ## 11. 如何添加您的模型
 <div id="lite.ai.toolkit-Contribute"></div>  
 
-如何添加您自己的模型以及成为贡献者？具体步骤请参考 [CONTRIBUTING.zh.md](docs/contrib/CONTRIBUTING.zh.md) ，或者，❤️不妨给个⭐️🌟star，这应该是最简单的支持方式了。
+如何添加您自己的模型以及成为贡献者？具体步骤请参考 [CONTRIBUTING.zh.md](https://github.com/DefTruth/lite.ai.toolkit/issues/191) ，或者，❤️不妨给个⭐️🌟star，这应该是最简单的支持方式了。
 
diff --git a/pylitex/.gitignore b/pylitex/.gitignore
@@ -0,0 +1,3 @@
+.DS_Store
+__pycache__
+.idea
diff --git a/pylitex/MANIFEST.in b/pylitex/MANIFEST.in
@@ -0,0 +1 @@
+recursive-include pylitex/hub/ *.onnx
diff --git a/pylitex/README.md b/pylitex/README.md
diff --git a/pylitex/docs/.gitignore b/pylitex/docs/.gitignore
diff --git a/pylitex/pylitex/__init__.py b/pylitex/pylitex/__init__.py
@@ -0,0 +1,2 @@
+# Versions
+__version__ = '0.0.1'
diff --git a/pylitex/pylitex/core/__init__.py b/pylitex/pylitex/core/__init__.py
diff --git a/pylitex/pylitex/cv/__init__.py b/pylitex/pylitex/cv/__init__.py
diff --git a/pylitex/pylitex/cv/face_landmarks_1000.py b/pylitex/pylitex/cv/face_landmarks_1000.py
diff --git a/pylitex/pylitex/cv/mg_matting.py b/pylitex/pylitex/cv/mg_matting.py
diff --git a/pylitex/pylitex/cv/nanodet.py b/pylitex/pylitex/cv/nanodet.py
diff --git a/pylitex/pylitex/cv/pfld106.py b/pylitex/pylitex/cv/pfld106.py
diff --git a/pylitex/pylitex/cv/pipnet.py b/pylitex/pylitex/cv/pipnet.py
diff --git a/pylitex/pylitex/cv/rvm.py b/pylitex/pylitex/cv/rvm.py
diff --git a/pylitex/pylitex/cv/scrfd.py b/pylitex/pylitex/cv/scrfd.py
diff --git a/pylitex/pylitex/cv/yolo5face.py b/pylitex/pylitex/cv/yolo5face.py
diff --git a/pylitex/pylitex/cv/yolov5.py b/pylitex/pylitex/cv/yolov5.py
diff --git a/pylitex/pylitex/cv/yolox.py b/pylitex/pylitex/cv/yolox.py
diff --git a/pylitex/pylitex/hub/.gitignore b/pylitex/pylitex/hub/.gitignore
@@ -0,0 +1 @@
+.DS_Store
diff --git a/pylitex/pylitex/lite.py b/pylitex/pylitex/lite.py
diff --git a/pylitex/pylitex/trackers/__init__.py b/pylitex/pylitex/trackers/__init__.py
diff --git a/pylitex/pylitex/types/__init__.py b/pylitex/pylitex/types/__init__.py
diff --git a/pylitex/pylitex/utils/__init__.py b/pylitex/pylitex/utils/__init__.py
diff --git a/pylitex/requirements.txt b/pylitex/requirements.txt
@@ -0,0 +1,9 @@
+# pylitex
+opencv-python>=4.5.2
+numpy>=1.14.4
+torch>=1.6.0
+onnxruntime>=1.7.0
+onnx>=1.8.0
+
+# test
+torchlm>=0.1.5
diff --git a/pylitex/setup.cfg b/pylitex/setup.cfg
@@ -0,0 +1,2 @@
+[metadata]
+desciption-file = README.md
diff --git a/pylitex/setup.py b/pylitex/setup.py
@@ -0,0 +1,42 @@
+import setuptools
+from pathlib import Path
+
+PATH_ROOT = Path(__file__).parent.resolve()
+
+
+def get_long_description():
+    description = (PATH_ROOT / "README.md").read_text(encoding="utf-8")
+    # replace relative repository path to absolute link to the release
+    static_url = f"https://github.com/DefTruth/lite.ai.toolkit/blob/main/pylitex"
+    description = description.replace("docs/res/", f"{static_url}/docs/res/")
+    return description
+
+
+setuptools.setup(
+    name="pylitex",
+    version="0.0.1",
+    author="DefTruth",
+    author_email="[email protected]",
+    description="A Python version of lite.ai.toolkit based on opencv and onnxruntime.",
+    long_description=get_long_description(),
+    long_description_content_type="text/markdown",
+    url="https://github.com/DefTruth/lite.ai.toolkit",
+    packages=setuptools.find_packages(),
+    install_requires=[
+        "opencv-python=4.5.2",
+        "numpy>=1.14.4",
+        "torch>=1.6.0",
+        "onnxruntime>=1.7.0",
+        "onnx>=1.8.0"
+    ],
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.6",
+        "Programming Language :: Python :: 3.7",
+        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent"
+    ],
+    include_package_data=True
+)
diff --git a/pylitex/test/.gitignore b/pylitex/test/.gitignore