debug counting in the system

src/roosts/system.py

@@ -163,8 +163,8 @@ def run_day_station(
 
         # save the list of tracks for UI, also save the list of sweeps and their animal counts
         self.visualizer.count_and_save(
-            cleaned_detections, tracks, self.count_cfg,
-            tracks_path, sweeps_path
+            cleaned_detections, tracks, self.postprocess.geosize, self.count_cfg,
+            self.dirs["scan_dir"], scanname2key, tracks_path, sweeps_path
         )
         delete_files([os.path.join(self.dirs["scan_dir"], key) for key in keys])
 

src/roosts/utils/counting_util.py

@@ -22,7 +22,7 @@ def xyr2geo(x, y, r, dim=600, rmax=150000, k=1.0):
     x0 = y0 = dim / 2.0  # origin
     x =  (x - x0) * 2 * rmax / dim
     y = -(y - y0) * 2 * rmax / dim
-    r = r * 2 * k * rmax / dim  # TODO: scaling by k may cause the r to be larger than the scope
+    r = r * k * 2 * rmax / dim  # TODO: scaling by k may cause the r to be larger than the scope
 
     return (x, y, r)
 

src/roosts/utils/visualizer.py

@@ -274,7 +274,10 @@ def save_gif(self, image_paths, outpath):
         imageio.mimsave(outpath, seq, "GIF", **kargs)
 
 
-    def count_and_save(self, detections, tracks, count_cfg, tracks_path, sweeps_path):
+    def count_and_save(
+        self, detections, tracks, geosize, count_cfg,
+        scan_dir, scanname2key, tracks_path, sweeps_path
+    ):
         """Save the list of tracks for UI, also save the list of sweeps and their animal counts"""
         det_dict = {}
         for det in detections:
@@ -303,7 +306,7 @@ def count_and_save(self, detections, tracks, count_cfg, tracks_path, sweeps_path
                     xyr = xyr2geo(
                         det["im_bbox"][0], det["im_bbox"][1], det["im_bbox"][2],
                         k=count_cfg["count_scaling"]
-                    )
+                    )  # geometric offset to radar
                     det["geo_dist"] = (xyr[0] ** 2 + xyr[1] ** 2) ** 0.5
 
                     f.write(
@@ -324,7 +327,7 @@ def count_and_save(self, detections, tracks, count_cfg, tracks_path, sweeps_path
                     with open(sweeps_path, 'a+') as ff:
                         # loop over sweeps, credit to Maria C. T. D. Belotti
                         radar = pyart.io.read_nexrad_archive(
-                            os.path.join(self.dirs["scan_dir"], scanname2key[det["scanname"]])
+                            os.path.join(scan_dir, scanname2key[det["scanname"]])
                         )
                         try:
                             sweep_indexes, sweep_angles = get_unique_sweeps(radar)
@@ -344,10 +347,10 @@ def count_and_save(self, detections, tracks, count_cfg, tracks_path, sweeps_path
                                 ff.write(
                                     ",".join([
                                         f"{det['track_ID']:d}", det["scanname"],
-                                        sweep_idx, f"{sweep_angle:.3f}",
+                                        f"{sweep_index}", f"{sweep_angle:.3f}",
 
                                         f"{count_cfg['count_scaling']:.3f}",
-                                        n_roost_pixels, n_overthresh_pixels, f"{n_animals:.3f}"
+                                        f"{n_roost_pixels}", f"{n_overthresh_pixels}", f"{n_animals:.3f}"
                                     ]) + "\n"
                                 )
                             except:

tools/launch_demo.py

@@ -10,14 +10,15 @@
 # TIMES = [(f"{year}0601", f"{year}1031") for year in range(2000, 2023)]
 # or
 STATIONS_TIMES = [
-    ("KTYX", "20200101", "20201231"),
-    ("KTYX", "20220101", "20221231"),
-    ("KLIX", "20200101", "20201231"),
-    ("KLIX", "20220101", "20221231"),
-    ("KDAX", "20200101", "20201231"),
-    ("KDAX", "20220101", "20221231"),
-    ("KTLX", "20200101", "20201231"),
-    ("KTLX", "20220101", "20221231"),
+    ("KTYX", "20200805", "20200806"),
+    # ("KTYX", "20200101", "20201231"),
+    # ("KTYX", "20220101", "20221231"),
+    # ("KLIX", "20200101", "20201231"),
+    # ("KLIX", "20220101", "20221231"),
+    # ("KDAX", "20200101", "20201231"),
+    # ("KDAX", "20220101", "20221231"),
+    # ("KTLX", "20200101", "20201231"),
+    # ("KTLX", "20220101", "20221231"),
 ]
 
 SPECIES = "swallow"