diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py index dade5999a..40d353c86 100644 --- 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 index 48b8aecbc..543a276fe 100644 --- 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 index 5286df9d0..c2ebc5112 100644 --- 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"] )