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()
