Spaces:
Paused
Paused
alessandro trinca tornidor
commited on
Commit
·
71ce930
1
Parent(s):
024790f
tests: fix broken test case because of missing pathlib import
Browse files
lisa_on_cuda/utils/app_helpers.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
import argparse
|
| 2 |
import logging
|
|
|
|
| 3 |
import os
|
| 4 |
import re
|
| 5 |
from typing import Callable
|
|
@@ -315,7 +316,7 @@ def get_inference_model_by_args(
|
|
| 315 |
|
| 316 |
def inference(
|
| 317 |
input_str: str,
|
| 318 |
-
input_image: str | np.ndarray,
|
| 319 |
internal_logger: logging = None,
|
| 320 |
embedding_key: str = None
|
| 321 |
):
|
|
@@ -357,8 +358,8 @@ def get_inference_model_by_args(
|
|
| 357 |
|
| 358 |
internal_logger.info("read and preprocess image.")
|
| 359 |
image_np = input_image
|
| 360 |
-
if isinstance(input_image, str):
|
| 361 |
-
image_np = cv2.imread(input_image)
|
| 362 |
image_np = cv2.cvtColor(image_np, cv2.COLOR_BGR2RGB)
|
| 363 |
original_size_list = [image_np.shape[:2]]
|
| 364 |
internal_logger.debug("start clip_image_processor.preprocess")
|
|
|
|
| 1 |
import argparse
|
| 2 |
import logging
|
| 3 |
+
from pathlib import Path
|
| 4 |
import os
|
| 5 |
import re
|
| 6 |
from typing import Callable
|
|
|
|
| 316 |
|
| 317 |
def inference(
|
| 318 |
input_str: str,
|
| 319 |
+
input_image: str | Path | np.ndarray,
|
| 320 |
internal_logger: logging = None,
|
| 321 |
embedding_key: str = None
|
| 322 |
):
|
|
|
|
| 358 |
|
| 359 |
internal_logger.info("read and preprocess image.")
|
| 360 |
image_np = input_image
|
| 361 |
+
if isinstance(input_image, str) or isinstance(input_image, Path):
|
| 362 |
+
image_np = cv2.imread(str(input_image))
|
| 363 |
image_np = cv2.cvtColor(image_np, cv2.COLOR_BGR2RGB)
|
| 364 |
original_size_list = [image_np.shape[:2]]
|
| 365 |
internal_logger.debug("start clip_image_processor.preprocess")
|
poetry.lock
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tests/imgs/{example1_mask_0.png → example0_mask_0.png}
RENAMED
|
File without changes
|
tests/imgs/{example1_mask_0_broken.png → example0_mask_0_broken.png}
RENAMED
|
File without changes
|
tests/test_app_helpers.py
CHANGED
|
@@ -87,3 +87,7 @@ class TestAppBuilders(unittest.TestCase):
|
|
| 87 |
logging.info(f"Written files with prefix '{prefix}' in {output_folder} folder.")
|
| 88 |
raise ae
|
| 89 |
logging.info("end")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 87 |
logging.info(f"Written files with prefix '{prefix}' in {output_folder} folder.")
|
| 88 |
raise ae
|
| 89 |
logging.info("end")
|
| 90 |
+
|
| 91 |
+
|
| 92 |
+
if __name__ == '__main__':
|
| 93 |
+
unittest.main()
|