diff --git a/fixtracks/utils/trackingdata.py b/fixtracks/utils/trackingdata.py index 3d0ca19..98a732a 100644 --- a/fixtracks/utils/trackingdata.py +++ b/fixtracks/utils/trackingdata.py @@ -19,6 +19,7 @@ class TrackingData(QObject): def setData(self, datadict): assert isinstance(datadict, dict) self._data = datadict + self._data["userlabeled"] = np.zeros_like(self["frame"], dtype=bool) self._columns = [k for k in self._data.keys()] @property @@ -81,6 +82,7 @@ class TrackingData(QObject): The new track id for the user-selected detections """ self._data["track"][self._user_selections] = track_id + self._data["userlabeled"][self._user_selections] = True def assignTracks(self, tracks): """assignTracks _summary_