Merge pull request #26 from DefTruth/dev

feat(PIPNet): add PIPNet to torchlm (#26)

Author: DefTruth

.gitignore

@@ -5,4 +5,5 @@ debug.py
 build
 dist
 torchlm.egg-info
-*.sh
+*.sh
+pretrained

MANIFEST.in

@@ -1 +1,2 @@
-recursive-include torchlm/transforms/assets *.jpg
+recursive-include torchlm/transforms/assets *.jpg
+recursive-include torchlm/tools/assets *.pth *.onnx

hubconf.py

@@ -0,0 +1,13 @@
+# Optional list of dependencies required by the package
+dependencies = ["torch", "torchvision"]
+
+from torchlm.models import (
+    pipnet_resnet18_10x68x32x256_300w,
+    pipnet_resnet101_10x68x32x256_300w,
+    pipnet_resnet18_10x19x32x256_aflw,
+    pipnet_resnet101_10x19x32x256_aflw,
+    pipnet_resnet18_10x29x32x256_cofw,
+    pipnet_resnet101_10x29x32x256_cofw,
+    pipnet_resnet18_10x98x32x256_wflw,
+    pipnet_resnet101_10x98x32x256_wflw
+)

requirements.txt

@@ -4,3 +4,6 @@ numpy>=1.14.4
 torch>=1.6.0
 torchvision>=0.9.0
 albumentations>=1.1.0
+onnx>=1.8.0
+onnxruntime>=1.7.0
+tqdm>=4.60.0

setup.py

@@ -14,7 +14,7 @@ def get_long_description():
 
 setuptools.setup(
     name="torchlm",
-    version="0.1.5",
+    version="0.1.6",
     author="DefTruth",
     author_email="qyjdef@163.com",
     description="A PyTorch landmarks-only library with 100+ data augmentations, "
@@ -28,8 +28,11 @@ def get_long_description():
         "opencv-python-headless>=4.5.2",
         "numpy>=1.14.4",
         "torch>=1.6.0",
-        "torchvision>=0.9.0",
-        "albumentations>=1.1.0"
+        "torchvision>=0.8.0",
+        "albumentations>=1.1.0",
+        "onnx>=1.8.0",
+        "onnxruntime>=1.7.0",
+        "tqdm>=4.60.0"
     ],
     classifiers=[
         "Programming Language :: Python :: 3",

test/.gitigore

@@ -1,2 +1,3 @@
 debug.py
-.DS_Store
+.DS_Store
+pretrained