Update main.py to reflect changes in neural_network.py

wjguan · web-flow · commit 014e193e4249 · 2024-05-24T10:58:09.000-04:00
diff --git a/DeepSlice/main.py b/DeepSlice/main.py
@@ -77,11 +77,11 @@ def predict(
         if use_secondary_model:
             print("Using secondary model")
             predictions = neural_network.predictions_util(
-                self.model, image_generator, secondary_weights,None, ensemble
+                self.model, image_generator, secondary_weights,None, ensemble, self.species
             )
         else:
             predictions = neural_network.predictions_util(
-                self.model, image_generator, primary_weights, secondary_weights, ensemble
+                self.model, image_generator, primary_weights, secondary_weights, ensemble, self.species
             )
         predictions["width"] = width
         predictions["height"] = height

Original file line number	Diff line number	Diff line change
`@@ -77,11 +77,11 @@ def predict(`
`77`	`77`	`if use_secondary_model:`
`78`	`78`	`print("Using secondary model")`
`79`	`79`	`predictions = neural_network.predictions_util(`
`80`		`- self.model, image_generator, secondary_weights,None, ensemble`
	`80`	`+ self.model, image_generator, secondary_weights,None, ensemble, self.species`
`81`	`81`	`)`
`82`	`82`	`else:`
`83`	`83`	`predictions = neural_network.predictions_util(`
`84`		`- self.model, image_generator, primary_weights, secondary_weights, ensemble`
	`84`	`+ self.model, image_generator, primary_weights, secondary_weights, ensemble, self.species`
`85`	`85`	`)`
`86`	`86`	`predictions["width"] = width`
`87`	`87`	`predictions["height"] = height`