Hayk Arutyunyan commited on
Commit
823433d
·
1 Parent(s): 599f938

fix: correct color ranges

Browse files
Files changed (1) hide show
  1. 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["from"], dtype=np.uint8)
25
- hi = np.array(rng["to"], dtype=np.uint8)
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. Оцифровка титула ----------