Vishakaraj
commited on
Commit
•
72a2e39
1
Parent(s):
72e9733
Debug handler
Browse files- handler.py +5 -5
handler.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import os
|
2 |
|
3 |
-
os.system("
|
4 |
os.system("pip install deepspeed==0.7.0")
|
5 |
|
6 |
from PIL import Image
|
@@ -10,10 +10,10 @@ import sys
|
|
10 |
import numpy as np
|
11 |
import torch
|
12 |
|
13 |
-
from detectron2.config import get_cfg
|
14 |
-
from detectron2.data.detection_utils import read_image
|
15 |
-
from detectron2.utils.logger import setup_logger
|
16 |
-
|
17 |
sys.path.insert(0, "third_party/CenterNet2/projects/CenterNet2/")
|
18 |
from centernet.config import add_centernet_config
|
19 |
from grit.config import add_grit_config
|
|
|
1 |
import os
|
2 |
|
3 |
+
os.system("pip install detectron2-0.6-cp39-cp39-linux_x86_64.whl")
|
4 |
os.system("pip install deepspeed==0.7.0")
|
5 |
|
6 |
from PIL import Image
|
|
|
10 |
import numpy as np
|
11 |
import torch
|
12 |
|
13 |
+
# from detectron2.config import get_cfg
|
14 |
+
# from detectron2.data.detection_utils import read_image
|
15 |
+
# from detectron2.utils.logger import setup_logger
|
16 |
+
print("###################################", os.getcwd())
|
17 |
sys.path.insert(0, "third_party/CenterNet2/projects/CenterNet2/")
|
18 |
from centernet.config import add_centernet_config
|
19 |
from grit.config import add_grit_config
|