jgrewe/fixtracks
1
0
Fork 0
Code Issues 4 Pull Requests Actions Packages Projects Releases Wiki Activity

[trackingdata] add column to mark user-labeled detections

Browse Source
This commit is contained in:
Jan Grewe 2025-02-18 09:18:32 +01:00
parent 6c46d834eb
commit 2737fed192
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fixtracks/utils/trackingdata.py
View File

@ -19,6 +19,7 @@ class TrackingData(QObject):
def setData(self, datadict): def setData(self, datadict):
assert isinstance(datadict, dict) assert isinstance(datadict, dict)
self._data = datadict self._data = datadict
self._data["userlabeled"] = np.zeros_like(self["frame"], dtype=bool)
self._columns = [k for k in self._data.keys()] self._columns = [k for k in self._data.keys()]
@property @property
@ -81,6 +82,7 @@ class TrackingData(QObject):
The new track id for the user-selected detections The new track id for the user-selected detections
""" """
self._data["track"][self._user_selections] = track_id self._data["track"][self._user_selections] = track_id
self._data["userlabeled"][self._user_selections] = True
def assignTracks(self, tracks): def assignTracks(self, tracks):
"""assignTracks _summary_ """assignTracks _summary_