use os.path.basename instead of splitting on forward slash

PolarBean · web-flow · commit 258f3b537396 · 2023-10-11T16:42:45.000+02:00
diff --git a/DeepSlice/neural_network/neural_network.py b/DeepSlice/neural_network/neural_network.py
@@ -170,7 +170,7 @@ def predictions_util(
         predictions = np.mean([predictions, secondary_predictions], axis=0)
         model.load_weights(primary_weights)
     filenames = image_generator.filenames
-    filenames = [i.split("/")[-1] for i in filenames]
+    filenames = [os.path.basename(i) for i in filenames]
     predictions_df = pd.DataFrame(
         {
             "Filenames": filenames,

Original file line number	Diff line number	Diff line change
`@@ -170,7 +170,7 @@ def predictions_util(`
`170`	`170`	`predictions = np.mean([predictions, secondary_predictions], axis=0)`
`171`	`171`	`model.load_weights(primary_weights)`
`172`	`172`	`filenames = image_generator.filenames`
`173`		`- filenames = [i.split("/")[-1] for i in filenames]`
	`173`	`+ filenames = [os.path.basename(i) for i in filenames]`
`174`	`174`	`predictions_df = pd.DataFrame(`
`175`	`175`	`{`
`176`	`176`	`"Filenames": filenames,`