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

cleanup

Browse Source
This commit is contained in:
Jan Grewe 2025-02-20 18:00:29 +01:00
parent 33b46af8d0
commit 20b2915b6b
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
fixtracks/fixtracks.py
View File

@ -1,11 +1,9 @@
from PySide6.QtCore import QSize, Qt
from PySide6.QtWidgets import QMainWindow, QWidget, QToolBar, QMenu, QMenuBar, QSizePolicy, QFileDialog
from PySide6.QtWidgets import QDialog, QVBoxLayout
from PySide6.QtGui import QKeySequence, QAction, QIcon, QPalette
from PySide6.QtWidgets import QMainWindow, QWidget, QToolBar, QSizePolicy, QFileDialog
from PySide6.QtGui import QKeySequence, QAction, QIcon
from fixtracks.widgets.centralwidget import CentralWidget
from fixtracks.dialogs.previewdialog import PreviewDialog
from fixtracks.utils.reader import ImageReader, DataFrameReader
from fixtracks.dialogs.about import AboutDialog
from fixtracks.dialogs.help import HelpDialog
import fixtracks.resources

4
fixtracks/widgets/classifier.py
View File

@ -239,7 +239,7 @@ class SizeClassifier(QWidget):
def setCoordinates(self, coordinates):
self._coordinates = coordinates
self._sizes = self.estimate_length(coordinates)
self._sizes = self.estimate_length(coordinates, bodyaxis=[0, 1, 2, 5])
n, e = self.estimate_histogram(self._sizes)
plot = self._plot_widget.addPlot()
bgi = pg.BarGraphItem(x0=e[:-1], x1=e[1:], height=n, pen='w', brush=(0,0,255,150))
@ -600,6 +600,7 @@ def main():
df = pickle.load(f)
data = TrackingData()
data.setData(as_dict(df))
coords = data.coordinates()
app = QApplication([])
window = QWidget()
@ -610,6 +611,7 @@ def main():
# else:
w = ClassifierWidget()
w.setData(data)
w.size_classifier.setCoordinates(coords)
layout = QVBoxLayout()
layout.addWidget(w)