Spaces:
Runtime error
Runtime error
nyonyong
commited on
Commit
·
0d81f48
1
Parent(s):
1b2702a
Directory fix
Browse files- TextDetection/utils/general.py +9 -0
- requirements.txt +1 -1
TextDetection/utils/general.py
CHANGED
|
@@ -35,6 +35,15 @@ import pkg_resources as pkg
|
|
| 35 |
import torch
|
| 36 |
import torchvision
|
| 37 |
import yaml
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
from ultralytics.yolo.utils.checks import check_requirements
|
| 39 |
|
| 40 |
from utils import TryExcept, emojis
|
|
|
|
| 35 |
import torch
|
| 36 |
import torchvision
|
| 37 |
import yaml
|
| 38 |
+
|
| 39 |
+
try:
|
| 40 |
+
import ultralytics
|
| 41 |
+
|
| 42 |
+
assert hasattr(ultralytics, "__version__") # verify package is not directory
|
| 43 |
+
except (ImportError, AssertionError):
|
| 44 |
+
os.system("pip install -U ultralytics")
|
| 45 |
+
import ultralytics
|
| 46 |
+
|
| 47 |
from ultralytics.yolo.utils.checks import check_requirements
|
| 48 |
|
| 49 |
from utils import TryExcept, emojis
|
requirements.txt
CHANGED
|
@@ -14,7 +14,7 @@ thop>=0.1.1 # FLOPs computation
|
|
| 14 |
torch>=1.7.0 # see https://pytorch.org/get-started/locally (recommended)
|
| 15 |
torchvision>=0.8.1
|
| 16 |
tqdm>=4.64.0
|
| 17 |
-
ultralytics>=8.0.
|
| 18 |
# protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012
|
| 19 |
|
| 20 |
# Logging ---------------------------------------------------------------------
|
|
|
|
| 14 |
torch>=1.7.0 # see https://pytorch.org/get-started/locally (recommended)
|
| 15 |
torchvision>=0.8.1
|
| 16 |
tqdm>=4.64.0
|
| 17 |
+
ultralytics>=8.0.232
|
| 18 |
# protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012
|
| 19 |
|
| 20 |
# Logging ---------------------------------------------------------------------
|