Merge pull request #3 from ilian6806/develop

Fixed dlib version. Cmake added to setup.

README.md

@@ -32,8 +32,8 @@ Use *Install from URL* option with this repo url.
 - mmdet==2.27.0
 - lightning-utilities==0.4.0
 - pytorch-lightning==1.7.6
+- dlib==19.24.0
 - setuptools
-- dlib
 
 All requirements will be installed on first use, except for:
  - [cmake](https://cmake.org/download/) - used only for dlib masks

install.py

@@ -40,9 +40,20 @@ python = sys.executable
 run(f'"{python}" -m pip install lightning-utilities==0.4.0', desc=None, errdesc=f"Couldn't install lightning-utilities")
 run(f'"{python}" -m pip install pytorch-lightning==1.7.6', desc=None, errdesc=f"Couldn't install pytorch-lightning")
 
+if not is_installed("cmake"):
+    run(f'"{python}" -m pip install cmake', desc="Installing cmake", errdesc="Couldn't install cmake")
+
 if not is_installed("dlib"):
     run(f'"{python}" -m pip install setuptools', desc="Installing setuptools", errdesc="Couldn't install setuptools")
-    run(f'"{python}" -m pip install dlib', desc="Installing dlib", errdesc="Couldn't install dlib")
+    try:
+        run(f'"{python}" -m pip install dlib==19.24.0', desc="Installing dlib", errdesc="Couldn't install dlib")
+    except Exception as e:
+        print("----------------------------------------")
+        print("Failed building wheel for dlib")
+        print("ERROR: CMake must be installed to build dlib")
+        print("Install cmake from https://cmake.org/download/")
+        print("----------------------------------------")
+
 
 if not is_installed("mmdet"):
     run(f'"{python}" -m pip install openmim==0.3.5', desc="Installing openmim", errdesc="Couldn't install openmim")