Spaces:
Paused
Paused
alessandro trinca tornidor
commited on
Commit
·
9eb0596
1
Parent(s):
f182d7a
[fix] fix wrong path for image placeholders folder
Browse files- utils/utils.py +5 -2
utils/utils.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
from enum import Enum
|
| 2 |
from pathlib import Path
|
| 3 |
|
|
@@ -169,8 +170,10 @@ def dict_to_cuda(input_dict):
|
|
| 169 |
def create_placeholder_variables():
|
| 170 |
import cv2
|
| 171 |
|
| 172 |
-
|
| 173 |
-
|
|
|
|
|
|
|
| 174 |
return {
|
| 175 |
"no_seg_out": no_seg_out,
|
| 176 |
"error_happened": error_happened
|
|
|
|
| 1 |
+
import logging
|
| 2 |
from enum import Enum
|
| 3 |
from pathlib import Path
|
| 4 |
|
|
|
|
| 170 |
def create_placeholder_variables():
|
| 171 |
import cv2
|
| 172 |
|
| 173 |
+
placeholders_folder = ROOT / "resources" / "placeholders"
|
| 174 |
+
logging.info(f"placeholders_folder:{placeholders_folder}.")
|
| 175 |
+
no_seg_out = cv2.imread(str(placeholders_folder / "no_seg_out.png"))[:, :, ::-1]
|
| 176 |
+
error_happened = cv2.imread(str(placeholders_folder / "error_happened.png"))[:, :, ::-1]
|
| 177 |
return {
|
| 178 |
"no_seg_out": no_seg_out,
|
| 179 |
"error_happened": error_happened
|