Merge pull request #92 from ltindall/patch_siggen_label

Update label_siggen.py to match new annotation_utils.py

Author: ltindall

label_scripts/label_siggen.py

@@ -9,8 +9,8 @@
 # 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"],
+    "am": ["/data/siggen/am.sigmf-meta"],
+    "fm": ["/data/siggen/fm.sigmf-meta"],
 }
 
 
@@ -19,33 +19,31 @@
         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,
+            data_obj,
+            avg_window_len=1024,
             debug=False,
-            spectral_energy_threshold=0.95,
-            # force_threshold_db=-1,
+            bandwidth_estimation=False,
+            force_threshold_db=-200,
             overwrite=True,
-            min_bandwidth=1e2,
-            min_annotation_length=256,
             dc_block=True,
+            labels={"am": {"annotation_length": (1024, None)}},
         )
 
 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,
+            data_obj,
+            avg_window_len=1024,
             debug=False,
-            # spectral_energy_threshold=0.95,
-            force_threshold_db=-0.1,
+            bandwidth_estimation=False,
+            force_threshold_db=-200,
             overwrite=True,
-            min_bandwidth=1e2,
-            min_annotation_length=256,
             dc_block=True,
+            labels={
+                "fm": {
+                    "annotation_length": (1024, None),
+                }
+            },
         )