def build_model(self, inputs: tf.Tensor) -> List[tf.Tensor]: """Build model with inputs and labels and print out model stats.""" logging.info('start building model') cls_outputs, box_outputs = inference.build_model( self.model_name, inputs, **self.model_config) # Write to tfevent for tensorboard. train_writer = tf.summary.FileWriter(self.logdir) train_writer.add_graph(tf.get_default_graph()) train_writer.flush() all_outputs = list(cls_outputs.values()) + list(box_outputs.values()) return all_outputs
def build_model(self, inputs: tf.Tensor, is_training: bool = False) -> List[tf.Tensor]: """Build model with inputs and labels and print out model stats.""" logging.info('start building model') cls_outputs, box_outputs = inference.build_model( self.model_name, inputs, is_training_bn=is_training, config=self.model_config) print('backbone+fpn+box params/flops = {:.6f}M, {:.9f}B'.format( *utils.num_params_flops())) # Write to tfevent for tensorboard. train_writer = tf.summary.FileWriter(self.logdir) train_writer.add_graph(tf.get_default_graph()) train_writer.flush() all_outputs = list(cls_outputs.values()) + list(box_outputs.values()) return all_outputs
if panel is None: panel = tk.Label(root, image=detected) panel.image = detected panel.pack(side="bottom") else: panel.configure(image=detected) panel.image = detected if is_fod: messagebox.showwarning(title="Detekce", message="Pozor, ve scéně se nachází cizí objekt!") if not is_fod: messagebox.showinfo(title="Detekce", message="Ve scéně se nenachází žádný cizí objekt.") root = tk.Tk(screenName="Foreign Object Detection") root.title("Detekce cizích objektů") detect_button = tk.Button(root, text="Detekovat cizí objekt", font=("Verdana", 10), width=25, bg="SkyBlue1", activebackground="white", command=detect) detect_button.pack() panel = None if __name__ == "__main__": model = build_model() root.mainloop()