Hayk Arutyunyan commited on
Commit ·
823433d
1
Parent(s): 599f938
fix: correct color ranges
Browse files- src/pipeline_hf.py +2 -2
src/pipeline_hf.py
CHANGED
|
@@ -21,8 +21,8 @@ def process_single_image(img: np.ndarray, cfg_path: str | Path = "configs/config
|
|
| 21 |
color_ranges = []
|
| 22 |
|
| 23 |
for key, rng in color_ranges_cfg.items():
|
| 24 |
-
lo = np.array(rng[
|
| 25 |
-
hi = np.array(rng[
|
| 26 |
color_ranges.append((lo, hi))
|
| 27 |
|
| 28 |
# ---------- 1. Оцифровка титула ----------
|
|
|
|
| 21 |
color_ranges = []
|
| 22 |
|
| 23 |
for key, rng in color_ranges_cfg.items():
|
| 24 |
+
lo = np.array(rng[0], dtype=np.uint8)
|
| 25 |
+
hi = np.array(rng[1], dtype=np.uint8)
|
| 26 |
color_ranges.append((lo, hi))
|
| 27 |
|
| 28 |
# ---------- 1. Оцифровка титула ----------
|