[trackingdata] add column to mark user-labeled detections

This commit is contained in:
Jan Grewe 2025-02-18 09:18:32 +01:00
parent 6c46d834eb
commit 2737fed192

View File

@ -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_