Added command line interface via fire library

cabelo · May 8, 2022 · a7cbf6d · a7cbf6d
1 parent e28d900
commit a7cbf6d
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py
@@ -12,6 +12,8 @@
 from tqdm import tqdm
 import pickle
 
+import fire
+
 from deepface.basemodels import VGGFace, OpenFace, Facenet, Facenet512, FbDeepFace, DeepID, DlibWrapper, ArcFace, Boosting
 from deepface.extendedmodels import Age, Gender, Race, Emotion
 from deepface.commons import functions, realtime, distance as dst
@@ -822,3 +824,4 @@ def detectFace(img_path, target_size = (224, 224), detector_backend = 'opencv',
 #main
 
 functions.initialize_folder()
+fire.Fire()
diff --git a/requirements.txt b/requirements.txt
@@ -13,3 +13,4 @@ lightgbm>=2.3.1
 dlib>=19.20.0
 retina-face>=0.0.1
 mediapipe>=0.8.7.3
+fire>=0.4.0
diff --git a/setup.py b/setup.py
@@ -19,6 +19,10 @@
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent"
     ],
+    entry_points={
+        "console_scripts":
+        ["DeepFace = deepface:DeepFace"],
+    },    
     python_requires='>=3.5.5',
-    install_requires=["numpy>=1.14.0", "pandas>=0.23.4", "tqdm>=4.30.0", "gdown>=3.10.1", "Pillow>=5.2.0", "opencv-python>=3.4.4", "tensorflow>=1.9.0", "keras>=2.2.0", "Flask>=1.1.2", "mtcnn>=0.1.0", "retina-face>=0.0.1"]
+    install_requires=["numpy>=1.14.0", "pandas>=0.23.4", "tqdm>=4.30.0", "gdown>=3.10.1", "Pillow>=5.2.0", "opencv-python>=3.4.4", "tensorflow>=1.9.0", "keras>=2.2.0", "Flask>=1.1.2", "mtcnn>=0.1.0", "retina-face>=0.0.1", "fire>=0.4.0"]
 )