Denoised and contrast amplified variants

liebharc · Jun 13, 2024 · acef9e2 · acef9e2
1 parent 94d337b
commit acef9e2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/homr/staff_parsing_tromr.py b/homr/staff_parsing_tromr.py
@@ -30,7 +30,12 @@ def apply_clahe(staff_image: NDArray, clip_limit: float = 2.0, kernel_size: int
 
 
 def build_image_options(staff_image: NDArray) -> list[NDArray]:
-    return [staff_image]
+    denoised1 = cv2.fastNlMeansDenoisingColored(staff_image, None, 10, 10, 7, 21)
+    return [
+        staff_image,
+        denoised1,
+        apply_clahe(denoised1),
+    ]
 
 
 def predict_best(

diff --git a/homr/transformer/decoder.py b/homr/transformer/decoder.py
@@ -242,7 +242,7 @@ def generate(  # noqa: PLR0915
                 if is_eos == 0:
                     break
                 retry = merger.add_symbol(rhythm_token[0][0], pitch_token[0][0], lift_token[0][0])
-                current_temperature *= 2
+                current_temperature *= 3.5
                 attempt += 1
 
             out_lift = torch.cat((out_lift, lift_sample), dim=-1)