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

[tracks] minor cleanup

Browse Source
This commit is contained in:
Jan Grewe 2025-02-03 20:53:42 +01:00
parent ee97382bbe
commit e2ac4f3551
2 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
fixtracks/widgets/detectionview.py
View File

@ -19,12 +19,6 @@ class Detection(QGraphicsEllipseItem):
def mousePressEvent(self, event): def mousePressEvent(self, event):
self.signals.clicked.emit(self.data(0), QPointF(event.scenePos().x(), event.scenePos().y())) self.signals.clicked.emit(self.data(0), QPointF(event.scenePos().x(), event.scenePos().y()))
# item = self.scene().itemAt(event.scenePos(), self.views()[0].transform())
# if item:
# item.setSelected(True)
# print(f"Rectangle clicked at: {event.scenePos().x()}, {event.scenePos().y()}")
def hoverEnterEvent(self, event): def hoverEnterEvent(self, event):
self.signals.hover.emit(self.data(0), QPointF(event.scenePos().x(), event.scenePos().y())) self.signals.hover.emit(self.data(0), QPointF(event.scenePos().x(), event.scenePos().y()))
@ -53,7 +47,6 @@ class DetectionScene(QGraphicsScene):
def mouseMoveEvent(self, event): def mouseMoveEvent(self, event):
if self.selection_rect is not None: if self.selection_rect is not None:
# Update the selection rectangle as the mouse moves
rect = QRectF(self.start_point, event.scenePos()).normalized() rect = QRectF(self.start_point, event.scenePos()).normalized()
self.selection_rect.setRect(rect) self.selection_rect.setRect(rect)
super().mouseMoveEvent(event) super().mouseMoveEvent(event)
@ -64,7 +57,6 @@ class DetectionScene(QGraphicsScene):
self.removeItem(self.selection_rect) self.removeItem(self.selection_rect)
self.selection_rect = None self.selection_rect = None
if rect.width() > 0.0: if rect.width() > 0.0:
# Find all items that intersect with the selection rectangle
selected_items = self.items(rect, Qt.IntersectsItemShape) selected_items = self.items(rect, Qt.IntersectsItemShape)
for item in selected_items: for item in selected_items:
if not isinstance(item, Detection): if not isinstance(item, Detection):
@ -158,8 +150,8 @@ class DetectionView(QWidget):
def main(): def main():
def items_selected(items): def items_selected(items):
print("items selected", items) print("items selected")
import pickle import pickle
import numpy as np import numpy as np
from IPython import embed from IPython import embed
@ -178,7 +170,7 @@ def main():
bg_coords = np.stack(df.keypoints[(df.track != 1) & (df.track != 2)].values,).astype(np.float32)[:,0,:] bg_coords = np.stack(df.keypoints[(df.track != 1) & (df.track != 2)].values,).astype(np.float32)[:,0,:]
bg_tracks = df.track[(df.track != 1) & (df.track != 2)].values bg_tracks = df.track[(df.track != 1) & (df.track != 2)].values
bg_ids = df.track[(df.track != 1) & (df.track != 2)].index.values bg_ids = df.track[(df.track != 1) & (df.track != 2)].index.values
scnd_coords = np.stack(df.keypoints[(df.track == 2)].values,).astype(np.float32)[:,0,:] scnd_coords = np.stack(df.keypoints[(df.track == 2)].values,).astype(np.float32)[:,0,:]
scnd_tracks = df.track[df.track == 2].values scnd_tracks = df.track[df.track == 2].values
scnd_ids = df.track[(df.track == 2)].index.values scnd_ids = df.track[(df.track == 2)].index.values

2
fixtracks/widgets/tracks.py
View File

@ -472,7 +472,7 @@ class FixTracks(QWidget):
self._data_combo.addItem("Please select") self._data_combo.addItem("Please select")
self._data_combo.addItems(self.fileList) self._data_combo.addItems(self.fileList)
self._data_combo.setCurrentIndex(0) self._data_combo.setCurrentIndex(0)
def populateKeypointCombo(self, num_keypoints): def populateKeypointCombo(self, num_keypoints):
self._keypointcombo.clear() self._keypointcombo.clear()
for i in range(num_keypoints): for i in range(num_keypoints):