Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ os.system('!git clone -b add_dit_inference_bis https://github.com/NielsRogge/uni
|
|
4 |
|
5 |
import argparse
|
6 |
import cv2
|
7 |
-
from ditod import add_vit_config
|
8 |
from detectron2.config import get_cfg
|
9 |
from detectron2.utils.visualizer import ColorMode, Visualizer
|
10 |
from detectron2.data import MetadataCatalog
|
|
|
4 |
|
5 |
import argparse
|
6 |
import cv2
|
7 |
+
from unilm.dit.object_detection.ditod import add_vit_config
|
8 |
from detectron2.config import get_cfg
|
9 |
from detectron2.utils.visualizer import ColorMode, Visualizer
|
10 |
from detectron2.data import MetadataCatalog
|