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

[detectionview] fix use of enum

Browse Source
This commit is contained in:
Jan Grewe 2025-02-06 10:45:31 +01:00
parent 2f40879624
commit 373671d237
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
fixtracks/widgets/detectionview.py
View File

@ -1,3 +1,4 @@
import enum
import logging
import numpy as np
@ -8,6 +9,14 @@ from PySide6.QtGui import QPixmap, QBrush, QColor, QImage
from fixtracks.info import PACKAGE_ROOT
from fixtracks.utils.signals import DetectionSignals, DetectionViewSignals, DetectionSceneSignals
class DetectionData(enum.Enum):
ID = 0
FRAME = 1
COORDINATES = 2
TRACK_ID = 3
class Detection(QGraphicsEllipseItem):
signals = DetectionSignals()
@ -130,10 +139,10 @@ class DetectionView(QWidget):
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.setData(DetectionData.TRACK_ID.value, track_ids[i])
item.setData(DetectionData.ID.value, detection_ids[i])
item.setData(DetectionData.COORDINATES.value, coordinates[i, :, :])
item.setData(DetectionData.FRAME.value, frames[i])
item = self._scene.addItem(item)
logging.debug("DetectionView: Number of items in scene: %i", len(self._scene.items()))