| import subprocess | |
| import sys | |
| import os | |
| # Install torch first | |
| subprocess.check_call([sys.executable, "-m", "pip", "install", "torch>=2.0.0", "torchvision>=0.15.0"]) | |
| # Clone and install detectron2 | |
| if not os.path.exists('detectron2'): | |
| subprocess.check_call(["git", "clone", "https://github.com/facebookresearch/detectron2"]) | |
| # Install detectron2 dependencies | |
| import distutils.core | |
| dist = distutils.core.run_setup("./detectron2/setup.py") | |
| deps = ' '.join([f"'{x}'" for x in dist.install_requires if 'torch' not in x]) | |
| subprocess.check_call(f"{sys.executable} -m pip install {deps}", shell=True) | |
| # Add detectron2 to path | |
| sys.path.insert(0, os.path.abspath('./detectron2')) | |