def next_file(self): self.__file_number += 1 if self.__file_number == len(self.__file_list) - 1: wx.PostEvent(self.__gui, ResultEvent(Events.EVT_NEXT_DONE_ID, 1)) if self.__file_number == len(self.__file_list): wx.PostEvent(self.__gui, ResultEvent(Events.EVT_NEXT_DONE_ID, 0)) self.__file_number = 0 try: self.__file_path = self.__file_list[self.__file_number] except ImportError: pass
def run(self): while self.__running: if self.__start: try: img = cv2.imread(self.__file_path) if self.__enable: img_res = processing_image(img) else: img_res = img if self.__save: self._save_image(self._get_save_path(self.__file_path), img_res) else: pass height, width = img_res.shape[:2] img_res = cv2.cvtColor(img_res, cv2.COLOR_BGR2RGB) bmp = wx.ImageFromBuffer(width, height, img_res) wx.PostEvent(self.__gui, ResultEvent(Events.EVT_DISPLAY_ID, bmp)) except Exception: pass self.Stop() else: pass time.sleep(0.1)
def run(self): while self.__running: if self.__start: if self.__cap.isOpened(): ret, frame = self.__cap.read() if ret: if self.__enable: frame_res = processing_image(frame) else: frame_res = frame if self.__save: self.__out.write(frame_res) else: pass height, width = frame_res.shape[:2] frame_res = cv2.cvtColor(frame_res, cv2.COLOR_BGR2RGB) bmp = wx.ImageFromBuffer(width, height, frame_res) wx.PostEvent(self.__gui, ResultEvent(Events.EVT_DISPLAY_ID, bmp)) else: self.Stop() else: pass time.sleep(1 / int(self.__fps)) else: time.sleep(0.1)
def Stop(self): self.__start = False wx.PostEvent(self.__gui, ResultEvent(Events.EVT_START_STATE_ID, 0))
def Start(self): self.__start = True wx.PostEvent(self.__gui, ResultEvent(Events.EVT_START_STATE_ID, 1))
def Stop(self): self._stop_out() self._stop_cap() wx.PostEvent(self.__gui, ResultEvent(Events.EVT_START_STATE_ID, 0)) self.__start = False
def Start(self): self._start_cap() if self.__save: self._start_out() wx.PostEvent(self.__gui, ResultEvent(Events.EVT_START_STATE_ID, 1)) self.__start = True