Removing the graph generation of the neural network

Thiago Franco de Moraes
1 parent 9e2fd40e
Showing 1 changed file with 0 additions and 2 deletions Show diff stats
invesalius/segmentation/brain/segment.py
@@ -74,8 +74,6 @@ def brain_segment(image, probability_array, comm_array):
         model = keras.models.model_from_json(json_file.read())
     model.load_weights(str(folder.joinpath("model.h5")))
     model.compile("Adam", "binary_crossentropy")
-    
-    keras.utils.plot_model(model, "model.png", show_shapes=True, show_layer_names=True)
  
     image = imagedata_utils.image_normalize(image, 0.0, 1.0, output_dtype=np.float32)
     sums = np.zeros_like(image)
...	...	@@ -74,8 +74,6 @@ def brain_segment(image, probability_array, comm_array):
74	74	model = keras.models.model_from_json(json_file.read())
75	75	model.load_weights(str(folder.joinpath("model.h5")))
76	76	model.compile("Adam", "binary_crossentropy")
77		-
78		- keras.utils.plot_model(model, "model.png", show_shapes=True, show_layer_names=True)
79	77
80	78	image = imagedata_utils.image_normalize(image, 0.0, 1.0, output_dtype=np.float32)
81	79	sums = np.zeros_like(image)
...	...