Expose colormap option in scanner

IQTLabs · Dec 20, 2023 · 8a99f49 · 8a99f49
1 parent 5768e23
commit 8a99f49
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/gamutrf/grscan.py b/gamutrf/grscan.py
@@ -27,6 +27,7 @@ class grscan(gr.top_block):
     def __init__(
         self,
         bucket_range=1.0,
+        colormap=16,
         compass=False,
         db_clamp_ceil=50,
         db_clamp_floor=-200,
@@ -217,7 +218,7 @@ def __init__(
                         norm_alpha=0,
                         norm_beta=1,
                         norm_type=32,  # cv::NORM_MINMAX = 32
-                        colormap=16,  # cv::COLORMAP_VIRIDIS = 16, cv::COLORMAP_TURBO = 20,
+                        colormap=colormap,  # cv::COLORMAP_VIRIDIS = 16, cv::COLORMAP_TURBO = 20,
                         interpolation=1,  # cv::INTER_LINEAR = 1,
                         flip=0,  # 0 means flipping around the x-axis
                         min_peak_points=inference_min_db,

diff --git a/gamutrf/scan.py b/gamutrf/scan.py
@@ -380,6 +380,14 @@ def argument_parser():
         default=0,
         help="if > 1, only send 1/n images for inference",
     )
+    parser.add_argument(
+        "--colormap",
+        dest="colormap",
+        type=int,
+        default=16,
+        help="""OpenCV colormap for spectrograms. See
+        https://docs.opencv.org/4.x/d3/d50/group__imgproc__colormap.html""",
+    )
     return parser