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
|