[tracks] minor cleanup
This commit is contained in:
parent
ee97382bbe
commit
e2ac4f3551
@ -19,12 +19,6 @@ class Detection(QGraphicsEllipseItem):
|
||||
|
||||
def mousePressEvent(self, event):
|
||||
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):
|
||||
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):
|
||||
if self.selection_rect is not None:
|
||||
# Update the selection rectangle as the mouse moves
|
||||
rect = QRectF(self.start_point, event.scenePos()).normalized()
|
||||
self.selection_rect.setRect(rect)
|
||||
super().mouseMoveEvent(event)
|
||||
@ -64,7 +57,6 @@ class DetectionScene(QGraphicsScene):
|
||||
self.removeItem(self.selection_rect)
|
||||
self.selection_rect = None
|
||||
if rect.width() > 0.0:
|
||||
# Find all items that intersect with the selection rectangle
|
||||
selected_items = self.items(rect, Qt.IntersectsItemShape)
|
||||
for item in selected_items:
|
||||
if not isinstance(item, Detection):
|
||||
@ -158,8 +150,8 @@ class DetectionView(QWidget):
|
||||
|
||||
def main():
|
||||
def items_selected(items):
|
||||
print("items selected", items)
|
||||
|
||||
print("items selected")
|
||||
|
||||
import pickle
|
||||
import numpy as np
|
||||
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_tracks = df.track[(df.track != 1) & (df.track != 2)].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_tracks = df.track[df.track == 2].values
|
||||
scnd_ids = df.track[(df.track == 2)].index.values
|
||||
|
@ -472,7 +472,7 @@ class FixTracks(QWidget):
|
||||
self._data_combo.addItem("Please select")
|
||||
self._data_combo.addItems(self.fileList)
|
||||
self._data_combo.setCurrentIndex(0)
|
||||
|
||||
|
||||
def populateKeypointCombo(self, num_keypoints):
|
||||
self._keypointcombo.clear()
|
||||
for i in range(num_keypoints):
|
||||
|
Loading…
Reference in New Issue
Block a user