add a densenet 2d

Kapoorlabs-CAPED · Dec 16, 2023 · 69044f7 · 69044f7
1 parent 4dfd53d
commit 69044f7
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/napatrackmater/Trackvector.py b/src/napatrackmater/Trackvector.py
@@ -463,9 +463,13 @@ def _final_morphological_dynamic_vectors(self, track_id):
                 self.unique_track_properties[track_id] = current_tracklets_properties
 
     def plot_mitosis_times(self, full_dataframe, save_path=""):
-        time_counter = Counter(self.cell_id_times)
-        times = list(time_counter.keys())
-        counts = list(time_counter.values())
+
+        dividing_events = full_dataframe[full_dataframe['Dividing'] == 1]
+
+        dividing_counts = dividing_events['t'].value_counts().sort_index()
+
+        times = dividing_counts.index
+        counts = dividing_counts.values
         data = {"Time": times, "Count": counts}
         df = pd.DataFrame(data)
         np.save(save_path + "_counts.npy", df.to_numpy())