def saveTest(): def startDetector(): detector.initMOG() detector.computeAll() tracker.trackAll(detector.detections) app = QtWidgets.QApplication(sys.argv) main_window = QtWidgets.QMainWindow() playback_manager = PlaybackManager(app, main_window) detector = Detector(playback_manager) tracker = Tracker(detector) fish_manager = FishManager(playback_manager, tracker) save_manager = SaveManager(playback_manager, detector, tracker, fish_manager) fish_manager.updateContentsSignal.connect( lambda: save_manager.saveFile(path, True)) playback_manager.openTestFile() detector._show_detections = True playback_manager.mapping_done.connect(startDetector) main_window.show() sys.exit(app.exec_())
def playbackTest(): """ Test code to assure tracker works with detector. """ def forwardImage(tuple): ind, frame = tuple detections = detector.getDetection(ind) image = cv2.applyColorMap(frame, cv2.COLORMAP_OCEAN) image = tracker.visualize(image, ind) figure.displayImage((ind, image)) def startDetector(): detector.initMOG() detector.computeAll() tracker.trackAll(detector.detections) playback_manager.play() app = QtWidgets.QApplication(sys.argv) main_window = QtWidgets.QMainWindow() playback_manager = PlaybackManager(app, main_window) detector = Detector(playback_manager) tracker = Tracker(detector) playback_manager.fps = 10 playback_manager.openTestFile() playback_manager.frame_available.connect(forwardImage) detector.mog_parameters.nof_bg_frames = 500 detector._show_detections = True playback_manager.mapping_done.connect(startDetector) figure = TestFigure(playback_manager.togglePlay) main_window.setCentralWidget(figure) LogObject().print(detector.parameters) LogObject().print(detector.parameters.mog_parameters) LogObject().print(tracker.parameters) main_window.show() sys.exit(app.exec_())
def dataTest(): def startDetector(): detector.initMOG() detector.computeAll() tracker.trackAll(detector.detections) app = QtWidgets.QApplication(sys.argv) main_window = Window() playback_manager = PlaybackManager(app, main_window) detector = Detector(playback_manager) tracker = Tracker(detector) fish_manager = FishManager(playback_manager, tracker) fish_list = FishList(fish_manager, playback_manager) playback_manager.openTestFile() detector.mog_parameters.nof_bg_frames = 500 detector._show_detections = True playback_manager.mapping_done.connect(startDetector) main_window.setCentralWidget(fish_list) main_window.show() sys.exit(app.exec_())