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

[tracks, detections] add frame and all keypoint coordinates to Detection

Browse Source
This commit is contained in:
Jan Grewe 2025-02-06 09:42:54 +01:00
parent a1353727c7
commit 6c75d24f97
2 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
fixtracks/widgets/detectionview.py
View File

@ -119,18 +119,21 @@ class DetectionView(QWidget):
self._scene.removeItem(it)
del it
def addDetections(self, coordinates:np.array, track_ids:np.array, detection_ids:np.array, brush:QBrush):
def addDetections(self, coordinates:np.array, track_ids:np.array, detection_ids:np.array, frames: np.array,
keypoint:int, brush:QBrush):
image_rect = self._pixmapitem.boundingRect() if self._pixmapitem is not None else QRectF(0,0,0,0)
num_detections = coordinates.shape[0]
for i in range(num_detections):
x = coordinates[i, 0]
y = coordinates[i, 1]
x = coordinates[i, keypoint, 0]
y = coordinates[i, keypoint, 1]
c = brush.color()
c.setAlpha(int(i * 255 / num_detections))
brush.setColor(c)
item = Detection(image_rect.left() + x, image_rect.top() + y, 20, 20, brush=brush)
item.setData(0, track_ids[i])
item.setData(1, detection_ids[i])
item.setData(2, coordinates[i, :, :])
item.setData(3, frames[i])
item = self._scene.addItem(item)
logging.debug("DetectionView: Number of items in scene: %i", len(self._scene.items()))

7
fixtracks/widgets/tracks.py
View File

@ -428,10 +428,11 @@ class FixTracks(QWidget):
if len(df) == 0:
return
keypoint = self._keypointcombo.currentIndex()
coords = np.stack(df["keypoints"].values).astype(np.float32)[:, keypoint,:]
coords = np.stack(df["keypoints"].values).astype(np.float32)[:, :,:]
tracks = df["track"].values.astype(int)
ids = df.index.values.astype(int)
self._detectionView.addDetections(coords, tracks, ids, self._brushes[name])
frames = df["frame"].values.astype(int)
self._detectionView.addDetections(coords, tracks, ids, frames, keypoint, self._brushes[name])
max_frames = self._data.max("frame")
start = self._timeline.rangeStart
@ -571,9 +572,11 @@ class FixTracks(QWidget):
logging.debug("Tracks: Detections selected")
tracks = np.zeros(len(detections))
ids = np.zeros_like(tracks)
coordinates = np.zeros((len(detections), ))
for i, d in enumerate(detections):
tracks[i] = d.data(0)
ids[i] = d.data(1)
self._data.setUserSelection(ids)
self._controls_widget.setSelectedTracks(tracks)
self.update()