Exemplo n.º 1
0
    def start(self):
        self.state = CaffeVisAppState(self.net, self.settings, self.bindings,
                                      self.net_layer_info)
        self.state.drawing_stale = True
        self.layer_print_names = [
            get_pretty_layer_name(self.settings, nn)
            for nn in self.state._layers
        ]

        if self.proc_thread is None or not self.proc_thread.is_alive():
            # Start thread if it's not already running
            self.proc_thread = CaffeProcThread(
                self.net, self.state, self.settings.caffevis_frame_wait_sleep,
                self.settings.caffevis_pause_after_keys,
                self.settings.caffevis_heartbeat_required,
                self.settings.caffevis_mode_gpu)
            self.proc_thread.start()

        if self.jpgvis_thread is None or not self.jpgvis_thread.is_alive():
            # Start thread if it's not already running
            self.jpgvis_thread = JPGVisLoadingThread(
                self.settings, self.state, self.img_cache,
                self.settings.caffevis_jpg_load_sleep,
                self.settings.caffevis_heartbeat_required)
            self.jpgvis_thread.start()