def __write_settings(self): """ Zapisz wartości z widgetów """ Configuration.distance_border1(self.border1_adjustment.get_value()) Configuration.distance_border2(self.border2_adjustment.get_value()) Configuration.meters_length( self.meters_length_spin_adjustment.get_value()) Configuration.color_number( self.color_number_spin_adjustment.get_value()) Configuration.horizontal_border( self.horizontal_border_spin_adjustment.get_value()) Configuration.vertical_border( self.vertical_border_spin_adjustment.get_value()) Configuration.pixel_limit(self.pixel_limit_spin_adjustment.get_value()) Configuration.distance_from_border( self.distance_from_border_spin_adjustment.get_value()) Configuration.draw_detection_region( self.draw_detection_region_check.get_active()) Configuration.draw_speed_region( self.draw_speed_region_check.get_active()) Configuration.draw_cars(self.draw_cars_check.get_active()) Configuration.draw_conturs(self.draw_conturs_check.get_active()) Configuration.draw_speed_info(self.draw_speed_info_check.get_active()) Configuration.draw_size_info(self.draw_size_info_check.get_active()) Configuration.draw_color_bar(self.draw_color_bar_check.get_active()) Configuration.save_config()
def __load_settings(self): """ Wczytaj obecną konfigurację znajdującą się config.json do widgetów okna. """ self.border1_adjustment.set_value(Configuration.distance_border1()) self.border2_adjustment.set_value(Configuration.distance_border2()) self.meters_length_spin_adjustment.set_value( Configuration.meters_length()) self.color_number_spin_adjustment.set_value( Configuration.color_number()) self.horizontal_border_spin_adjustment.set_value( Configuration.horizontal_border()) self.vertical_border_spin_adjustment.set_value( Configuration.vertical_border()) self.pixel_limit_spin_adjustment.set_value(Configuration.pixel_limit()) self.distance_from_border_spin_adjustment.set_value( Configuration.distance_from_border()) self.draw_detection_region_check.set_active( Configuration.draw_detection_region()) self.draw_speed_region_check.set_active( Configuration.draw_speed_region()) self.draw_cars_check.set_active(Configuration.draw_cars()) self.draw_conturs_check.set_active(Configuration.draw_conturs()) self.draw_speed_info_check.set_active(Configuration.draw_speed_info()) self.draw_size_info_check.set_active(Configuration.draw_size_info()) self.draw_color_bar_check.set_active(Configuration.draw_color_bar())
def __write_settings(self): """ Zapisz wartości z widgetów """ Configuration.distance_border1(self.border1_adjustment.get_value()) Configuration.distance_border2(self.border2_adjustment.get_value()) Configuration.meters_length(self.meters_length_spin_adjustment.get_value()) Configuration.color_number(self.color_number_spin_adjustment.get_value()) Configuration.horizontal_border(self.horizontal_border_spin_adjustment.get_value()) Configuration.vertical_border(self.vertical_border_spin_adjustment.get_value()) Configuration.pixel_limit(self.pixel_limit_spin_adjustment.get_value()) Configuration.distance_from_border(self.distance_from_border_spin_adjustment.get_value()) Configuration.draw_detection_region(self.draw_detection_region_check.get_active()) Configuration.draw_speed_region(self.draw_speed_region_check.get_active()) Configuration.draw_cars(self.draw_cars_check.get_active()) Configuration.draw_conturs(self.draw_conturs_check.get_active()) Configuration.draw_speed_info(self.draw_speed_info_check.get_active()) Configuration.draw_size_info(self.draw_size_info_check.get_active()) Configuration.draw_color_bar(self.draw_color_bar_check.get_active()) Configuration.save_config()
def __load_settings(self): """ Wczytaj obecną konfigurację znajdującą się config.json do widgetów okna. """ self.border1_adjustment.set_value(Configuration.distance_border1()) self.border2_adjustment.set_value(Configuration.distance_border2()) self.meters_length_spin_adjustment.set_value(Configuration.meters_length()) self.color_number_spin_adjustment.set_value(Configuration.color_number()) self.horizontal_border_spin_adjustment.set_value(Configuration.horizontal_border()) self.vertical_border_spin_adjustment.set_value(Configuration.vertical_border()) self.pixel_limit_spin_adjustment.set_value(Configuration.pixel_limit()) self.distance_from_border_spin_adjustment.set_value(Configuration.distance_from_border()) self.draw_detection_region_check.set_active(Configuration.draw_detection_region()) self.draw_speed_region_check.set_active(Configuration.draw_speed_region()) self.draw_cars_check.set_active(Configuration.draw_cars()) self.draw_conturs_check.set_active(Configuration.draw_conturs()) self.draw_speed_info_check.set_active(Configuration.draw_speed_info()) self.draw_size_info_check.set_active(Configuration.draw_size_info()) self.draw_color_bar_check.set_active(Configuration.draw_color_bar())
def perform(frame: Frame, database, img_saver, run_classyfication=True): """ Dokonuje przetwarzania ramki obrazu przez algorytm. :param frame: Ramka obrazu. :param database: Baza danych do zapisywania parametrów. :param img_saver: Obiekt zapisyjący pliki obrazów. :return: Przetworzona ramka. """ vehicles, mask = Detector.find_vehicles(frame) objects = Follower.update(vehicles, frame, mask) records = [] if run_classyfication: if objects is not None: for obj in objects: record = Classyfication.perform(obj, Algorithm.file) records.append(record) if database is not None: database.write(record, Algorithm.file) if img_saver is not None: img_saver.write(record, Algorithm.file) mask = cv2.cvtColor(mask, cv2.COLOR_GRAY2BGR) # Rysowanie pojazdów. if Configuration.draw_cars(): frame, mask = Detector.draw_vehicles(frame, mask, vehicles) # Rysowanie obszaru wykrywania. if Configuration.draw_detection_region(): frame, mask = Detector.draw_detection_region(frame, mask) # Rysowanie obszaru pomiaru prędkości if Configuration.draw_speed_region(): frame = Classyfication.draw_speed_region(frame) return frame, mask, records