🔨 Fix Detectron2 installation

affromero · Jun 8, 2023 · 56799ef · 56799ef
1 parent 7145cda
commit 56799ef
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/INSTALL.md b/INSTALL.md
@@ -34,8 +34,7 @@ We use an evironment with the following specifications, packages and dependencie
   pip3 install -U opencv-python
 
   # Install detectron2
-  python -m pip install detectron2 -f \
-    https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html
+  python tools/setup_detectron2.py
 
   # Install other dependencies
   pip3 install git+https://github.com/cocodataset/panopticapi.git

diff --git a/tools/setup_detectron2.py b/tools/setup_detectron2.py
@@ -0,0 +1,11 @@
+import sys, os, distutils.core, subprocess
+
+if not os.path.exists('./detectron2'):
+    subprocess.run(['git', 'clone', 'https://github.com/facebookresearch/detectron2'])
+
+dist = distutils.core.run_setup("./detectron2/setup.py")
+
+for x in dist.install_requires:
+    subprocess.run(['python', '-m', 'pip', 'install', x])
+
+sys.path.insert(0, os.path.abspath('./detectron2'))