logo
Browse Source

fix detectron2 install issue

main
zilliz 3 years ago
parent
commit
97caad9d9c
  1. 3
      detectron.py

3
detectron.py

@ -3,8 +3,7 @@ import os
try:
import detectron2
except ModuleNotFoundError:
os.system("git clone https://github.com/facebookresearch/detectron2.git")
os.system("python -m pip install -e detectron2")
os.system("pip install 'git+https://github.com/facebookresearch/detectron2.git'")
from detectron2 import model_zoo
from detectron2.config import get_cfg
from detectron2.engine.defaults import DefaultPredictor

Loading…
Cancel
Save