def __init__(self): self.path_manager = ds.PathManager() self.dset_store = ds.DetectionSetStore() self.hypothesis_manager = ds.HypothesisManager() self.claim_manager = ds.ClaimManager() self.graph = ds.Graph(self.dset_store, future_depth=3)
def __init__(self): self.dset_store = ds.DetectionSetStore() self.modules = [ vm.PathCongruence(), vm.PairsContinuity(), vm.EqualityCounter() ]
def __init__(self, app, parent=None): super(MainWindow, self).__init__(parent) self.resize(1000, 600) self.setWindowTitle('BeesBook Filtering Editor') self.central_widget = QtGui.QStackedWidget(self) self.setCentralWidget(self.central_widget) self.dset_store = ds.DetectionSetStore() self.path_manager = ds.PathManager() self.loader_tab = LoaderTab(self, app) self.editor_tab = EditorTab(self, app) self.central_widget.addWidget(self.loader_tab) self.central_widget.addWidget(self.editor_tab) self.central_widget.setCurrentWidget(self.loader_tab)
def __init__(self): self.path_manager = ds.PathManager() self.dset_store = ds.DetectionSetStore() self.claim_manager = ds.ClaimManager()
def load_data(self): if not os.path.exists(config.DATA_FOLDER): print 'Error: folder not found' return self.block_inputs(True) self.dset_store = ds.DetectionSetStore() self.path_manager = None self.paths_load_progress.setValue(0) self.paths_load_label.setText('') try: repo = Repository(config.DATA_FOLDER) start_time = datetime(config.DATE[0], config.DATE[1], config.DATE[2], config.TIME[0], config.TIME[1], tzinfo=pytz.utc) fnames = repo.iter_fnames(begin=start_time) for fname in fnames: frame_container = load_frame_container(fname) cam = frame_container.camId #frame_container.fromTimestamp # already available #frame_container.toTimestamp # already available self.dset_store.source = frame_container.dataSources[ 0].filename previous_timestamp = None self.data_load_progress.setMaximum(config.FRAME_END + 1 - config.FRAME_START) self.app.processEvents() frame_index = config.FRAME_START for frame in list(frame_container.frames )[config.FRAME_START:config.FRAME_END + 1]: #timestamp = frame.timestamp # not included yet #frame.id # not included yet timestamp = ds.TimeStamp(frame_index, cam) timestamp.connect_with_previous(previous_timestamp) previous_timestamp = timestamp dset = ds.DetectionSet() self.dset_store.store[timestamp] = dset data = convert_frame_to_numpy(frame) for detection_data in data: dset.add_detection( ds.Detection( detection_data['idx'], timestamp, np.array( [ detection_data['ypos'], detection_data['xpos'] ] ), # rotated, otherwise will be portrait orientation detection_data['localizerSaliency'], detection_data['decodedId'] [:: -1] # reversed, we want least significant bit last )) frame_index += 1 self.data_load_progress.setValue(frame_index - config.FRAME_START) self.app.processEvents() self.data_load_label.setText( str(len(self.dset_store.store)) + ' frames loaded') self.app.processEvents() # break because we only load the first fname break except: pass self.block_inputs(False)
def __init__(self): self.future_depth = 4 self.dset_store = ds.DetectionSetStore() self.graph = ds.Graph(self.dset_store, future_depth=self.future_depth) self.path_manager = PathManager(self.dset_store)
def __init__(self): self.dset_store = ds.DetectionSetStore()