zilliz
3 years ago
2 changed files with
6 additions and
8 deletions
-
__init__.py
-
detectron.py
|
|
|
@ -1,10 +1,3 @@ |
|
|
|
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") |
|
|
|
|
|
|
|
from .detectron import Detectron2 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -1,5 +1,10 @@ |
|
|
|
from typing import List, Tuple |
|
|
|
|
|
|
|
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") |
|
|
|
from detectron2 import model_zoo |
|
|
|
from detectron2.config import get_cfg |
|
|
|
from detectron2.engine.defaults import DefaultPredictor |
|
|
|
|
