resolve conflicts

Author: ltindall

custom_handlers/requirements.txt

@@ -0,0 +1 @@
+setuptools==69.5.1

experiments/mixed_experiments.py

@@ -1,3 +1,8 @@
+import torch
+
+torch.set_float32_matmul_precision("high")
+
+
 from rfml.experiment import *
 
 # Ensure that data directories have sigmf-meta files with annotations
@@ -230,6 +235,14 @@
         "spec_epochs": 0,
         "notes": "DJI Mavic3 Detection",
     },
+    "experiment_siggen": {
+        "class_list": ["am", "fm"],
+        "train_dir": ["/scratch/tmp/rfmltest"],
+        "iq_epochs": 10,
+        "spec_epochs": 0,
+        "notes": "narrowband test",
+        "iq_only_start_of_burst": False,
+    },
 }
 
 
@@ -255,6 +268,7 @@
         # "experiment_ettus_2",
         # "experiment_blade_2",
         # "experiment_mavic3",
+        "experiment_siggen",
     ]
 
     train({name: experiments[name] for name in experiments_to_run})

label_scripts/label_siggen.py

@@ -0,0 +1,51 @@
+import glob
+
+from pathlib import Path
+from tqdm import tqdm
+
+import rfml.annotation_utils as annotation_utils
+import rfml.data as data_class
+
+# generated with
+# $ for i in am fm ; do /scratch/iqtlabs/rfml/utils/siggen.py --samp_rate 1000000 --siggen $i --int_count 1000 ; done
+data_globs = {
+    "am": ["/scratch/tmp/rfmltest/am.sigmf-meta"],
+    "fm": ["/scratch/tmp/rfmltest/fm.sigmf-meta"],
+}
+
+
+for data_glob in data_globs["am"]:
+    for f in tqdm(glob.glob(str(Path(data_glob)))):
+        data_obj = data_class.Data(f)
+        annotation_utils.reset_annotations(data_obj)
+        annotation_utils.annotate(
+            f,
+            label="am",
+            avg_window_len=256,
+            avg_duration=0.10,
+            debug=False,
+            spectral_energy_threshold=0.95,
+            # force_threshold_db=-1,
+            overwrite=True,
+            min_bandwidth=1e2,
+            min_annotation_length=256,
+            dc_block=True,
+        )
+
+for data_glob in data_globs["fm"]:
+    for f in tqdm(glob.glob(str(Path(data_glob)))):
+        data_obj = data_class.Data(f)
+        annotation_utils.reset_annotations(data_obj)
+        annotation_utils.annotate(
+            f,
+            label="fm",
+            avg_window_len=256,
+            avg_duration=0.10,
+            debug=False,
+            # spectral_energy_threshold=0.95,
+            force_threshold_db=-0.1,
+            overwrite=True,
+            min_bandwidth=1e2,
+            min_annotation_length=256,
+            dc_block=True,
+        )

poetry.lock

@@ -11,11 +11,11 @@ scikit-image = "^0.24.0"
 matplotlib = "^3.8.3"
 numpy = "^1.26.4"
 opencv-python = "^4.8.0.74"
-scipy = "1.13"
+scipy = "1.13.1"
 zstandard = "^0.23.0"
 pyyaml = "^6.0.1"
 pillow = "^10.2.0"
-python-on-whales = "^0.72.0"
+python-on-whales = "^0.73.0"
 sigmf = "^1.2.0"
 tqdm = "^4.66.4"
 torch = "2.0.1"

pyproject.toml

@@ -0,0 +1,8 @@
+FROM ubuntu:24.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PATH="${PATH}:/root/.local/bin"
+RUN apt-get -yq update && apt-get install --no-install-recommends -yq festival festvox-us-slt-hts python3-pip gnuradio sox && pip install --break-system-packages sigmf
+COPY siggen.py /root/siggen.py
+RUN /root/siggen.py --int_count 1 --samp_rate 100000 --siggen am --sample_file /tmp/test && rm -f /tmp/test
+CMD ["/root/siggen.py"]