cleanup

fixtracks/widgets/classifier.py

@@ -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)